Hugging Face's logo

Spaces:
Jugal-sheth
/
MNIST-classifier
Sleeping

App Files Community
Jugal-sheth commited on
Commit
f2e4334
·
1 Parent(s): 90b6d22

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -4,8 +4,7 @@ from model import model
4
  from torchvision import transforms
5
 
6
  #Load the Model
7
- model.load_state_dict(torch.load('mnist_model.pth'), map_location=torch.device('cpu'))
8
-
9
  # Set model to evaluation mode
10
  model.eval()
11
 
 
4
  from torchvision import transforms
5
 
6
  #Load the Model
7
+ model.load_state_dict(torch.load('mnist_model.pth'))
 
8
  # Set model to evaluation mode
9
  model.eval()
10