athulnambiar
commited on
Commit
•
ce54ee1
1
Parent(s):
391ecac
Update app.py
Browse files
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(
|