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

re-corrected 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="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="model/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()