Spaces:

balakrish181
/

mnist

Sleeping

balakrish181 commited on May 26

Commit

34b3108

•

1 Parent(s): 4027be6

corrected error

Files changed (1) hide show

model.py CHANGED Viewed

@@ -47,7 +47,7 @@ def load_model():
     transforms.Grayscale(num_output_channels=1)
 ])
-    model.load_state_dict(torch.load('demos/mnist_app/best_model.pth',map_location='cpu'))
     return model,transforming,model.classes

     transforms.Grayscale(num_output_channels=1)
 ])
+    model.load_state_dict(torch.load('best_model.pth',map_location='cpu'))
     return model,transforming,model.classes