Jasur05 commited on
Commit
aa8c8ea
·
verified ·
1 Parent(s): e9d2555

corrected saved model name

Browse files
Files changed (1) hide show
  1. model.py +1 -1
model.py CHANGED
@@ -17,7 +17,7 @@ def create_model(num_classes=7):
17
 
18
  return model
19
 
20
- def load_model(weights_path="fine_tuned_effnetb2_dermamnist.pth"):
21
  model = create_model(num_classes=7)
22
  model.load_state_dict(torch.load(weights_path, map_location=torch.device("cpu")))
23
  model.eval()
 
17
 
18
  return model
19
 
20
+ def load_model(weights_path="effnetb2_dermamnist.pth"):
21
  model = create_model(num_classes=7)
22
  model.load_state_dict(torch.load(weights_path, map_location=torch.device("cpu")))
23
  model.eval()