update app.py

app.py

@@ -7,7 +7,7 @@ from torchvision.transforms import ToTensor
 # Define model
 class ConvNet(nn.Module):
     def __init__(self):
-        super(CNN, self).__init__()
+        super(ConvNet, self).__init__()
         self.conv1 = nn.Conv2d(1, 32, kernel_size=5)
         self.conv2 = nn.Conv2d(32, 32, kernel_size=5)
         self.conv3 = nn.Conv2d(32,64, kernel_size=5)