Update app.py
Browse files
app.py
CHANGED
@@ -51,14 +51,15 @@ class Cnn(nn.Module):
|
|
51 |
torch.manual_seed(0)
|
52 |
|
53 |
|
54 |
-
# Create an instance of your model
|
55 |
-
model = NeuralNetClassifier(
|
56 |
-
|
57 |
-
|
58 |
-
|
59 |
-
|
60 |
-
|
61 |
-
)
|
|
|
62 |
|
63 |
# Specify the path to the saved model weights
|
64 |
model_weights_path = 'model_weights.pth'
|
|
|
51 |
torch.manual_seed(0)
|
52 |
|
53 |
|
54 |
+
# # Create an instance of your model
|
55 |
+
# model = NeuralNetClassifier(
|
56 |
+
# Cnn,
|
57 |
+
# max_epochs=10,
|
58 |
+
# lr=0.002,
|
59 |
+
# optimizer=torch.optim.Adam,
|
60 |
+
# device=device,
|
61 |
+
# )
|
62 |
+
model=Cnn()
|
63 |
|
64 |
# Specify the path to the saved model weights
|
65 |
model_weights_path = 'model_weights.pth'
|