Update app.py

app.py

@@ -51,14 +51,15 @@ class Cnn(nn.Module):
 torch.manual_seed(0)
 
 
-# Create an instance of your model
-model = NeuralNetClassifier(
-    Cnn,
-    max_epochs=10,
-    lr=0.002,
-    optimizer=torch.optim.Adam,
-    device=device,
-)
+# # Create an instance of your model
+# model = NeuralNetClassifier(
+#     Cnn,
+#     max_epochs=10,
+#     lr=0.002,
+#     optimizer=torch.optim.Adam,
+#     device=device,
+# )
+model=Cnn()
 
 # Specify the path to the saved model weights
 model_weights_path = 'model_weights.pth'