Spaces:

Gosula
/

hand_written_digit_recognition

Sleeping

Gosula commited on Nov 4, 2023

Commit

30e24a4

•

1 Parent(s): d7be2f6

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -48,7 +48,7 @@ model = Cnn()
 model_weights_path = 'model_weights.pth'
 # Load the model weights
-model.load_state_dict(torch.load(model_weights_path),map_location=torch.device('cpu') )
 # Set the model to evaluation mode for inference
 model.eval()

 model_weights_path = 'model_weights.pth'
 # Load the model weights
+model.load_state_dict(torch.load(model_weights_path,map_location=torch.device('cpu')))
 # Set the model to evaluation mode for inference
 model.eval()