athulnambiar commited on
Commit
ce54ee1
1 Parent(s): 391ecac

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -65,7 +65,8 @@ def load_model():
65
  model = MelanomaModel(out_size=9)
66
 
67
  # Path to your model file
68
- model_path = os.path.join("model", "multi_weight.pth")
 
69
 
70
  # If you trust the checkpoint file, set weights_only=False
71
  checkpoint = torch.load(
 
65
  model = MelanomaModel(out_size=9)
66
 
67
  # Path to your model file
68
+ # model_path = os.path.join("model", "multi_weight.pth")
69
+ model_path = "multi_weight.pth"
70
 
71
  # If you trust the checkpoint file, set weights_only=False
72
  checkpoint = torch.load(