Spaces:
Sleeping
Sleeping
Commit
·
a864ac4
1
Parent(s):
5d1fb06
model path update
Browse files
app.py
CHANGED
|
@@ -20,7 +20,7 @@ def classify_image(img):
|
|
| 20 |
#initialize the model
|
| 21 |
loaded_model = MNISTnet(input_channels=1, num_labels=10, hidden_layers=5).eval()
|
| 22 |
#put the state dict values
|
| 23 |
-
model_state_dict_path = Path("
|
| 24 |
loaded_model.load_state_dict(torch.load(model_state_dict_path))
|
| 25 |
# make the prediction
|
| 26 |
with torch.inference_mode():
|
|
|
|
| 20 |
#initialize the model
|
| 21 |
loaded_model = MNISTnet(input_channels=1, num_labels=10, hidden_layers=5).eval()
|
| 22 |
#put the state dict values
|
| 23 |
+
model_state_dict_path = Path("models/MNISTnet_state_dict.pt")
|
| 24 |
loaded_model.load_state_dict(torch.load(model_state_dict_path))
|
| 25 |
# make the prediction
|
| 26 |
with torch.inference_mode():
|