Spaces:

MostafaAhmed98
/

arabic_letters_recognition

Sleeping

MostafaAhmed98 commited on Jun 5

Commit

e879ffa

•

1 Parent(s): 1f6acc0

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -35,7 +35,7 @@ def load_the_model():
         return x
   loaded_model = Net()#.to(device)
-  loaded_model.load_state_dict(torch.load(path_of_model), map_location=torch.device('cpu'))
   loaded_model.eval()
   return loaded_model

         return x
   loaded_model = Net()#.to(device)
+  loaded_model.load_state_dict(torch.load(path_of_model, map_location=torch.device('cpu')))
   loaded_model.eval()
   return loaded_model