Spaces:

arpandesign
/

NumberClassifier

Sleeping

arpandesign commited on Jul 4, 2025

Commit

6a4fc80

verified ·

1 Parent(s): 17836ed

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -72,7 +72,7 @@ model_0 = NumberClassifier(input_shape=1, # number of color channels (3 for RGB)
                   hidden_units=10,
                   output_shape=10).to(device)
-model_0.load_state_dict(torch.load("pytorch_num_classifier_final_model_with_EMNIST.pth", map_location=torch.device('cpu')))
 title = "Number Classifier Minimal"

                   hidden_units=10,
                   output_shape=10).to(device)
+model_0.load_state_dict(torch.load("models/pytorch_num_classifier_final_model_with_EMNIST.pth", map_location=torch.device('cpu')))
 title = "Number Classifier Minimal"