binaychandra commited on
Commit
a864ac4
·
1 Parent(s): 5d1fb06

model path update

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -20,7 +20,7 @@ def classify_image(img):
20
  #initialize the model
21
  loaded_model = MNISTnet(input_channels=1, num_labels=10, hidden_layers=5).eval()
22
  #put the state dict values
23
- model_state_dict_path = Path("/models/MNISTnet_state_dict.pt")
24
  loaded_model.load_state_dict(torch.load(model_state_dict_path))
25
  # make the prediction
26
  with torch.inference_mode():
 
20
  #initialize the model
21
  loaded_model = MNISTnet(input_channels=1, num_labels=10, hidden_layers=5).eval()
22
  #put the state dict values
23
+ model_state_dict_path = Path("models/MNISTnet_state_dict.pt")
24
  loaded_model.load_state_dict(torch.load(model_state_dict_path))
25
  # make the prediction
26
  with torch.inference_mode():