Spaces:

Rounak28
/

digit_recogniser

Sleeping

Rounak28 commited on Jun 27, 2023

Commit

faf1760

•

1 Parent(s): 375800a

jyfgyjsgf

Files changed (1) hide show

app.py CHANGED Viewed

@@ -10,7 +10,7 @@ from model import NeuralNet
 device = "cuda" if torch.cuda.is_available() else "cpu"
 model = NeuralNet().to(device)
-model.load_state_dict(torch.load("models/model.pth"))
 labels = [

 device = "cuda" if torch.cuda.is_available() else "cpu"
 model = NeuralNet().to(device)
+model.load_state_dict(torch.load("models/model.pth", map_location=torch.device('cpu')))
 labels = [