Mikecode123 commited on
Commit
801ed07
·
verified ·
1 Parent(s): 8b65e00

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -62,11 +62,11 @@ def load_model(path):
62
  # -----------------------------
63
  # MODEL PATHS
64
  # -----------------------------
65
- BASE = Path("saved_models")
66
 
67
- model_121 = load_model(BASE / "alzheimers_densenet121.pth")
68
- model_169 = load_model(BASE / "alzheimers_densenet169.pth")
69
- model_201 = load_model(BASE / "alzheimers_densenet201.pth")
 
70
 
71
  # -----------------------------
72
  # SINGLE MODEL PREDICTION
 
62
  # -----------------------------
63
  # MODEL PATHS
64
  # -----------------------------
 
65
 
66
+
67
+ model_121 = "alzheimers_densenet121.pth"
68
+ model_169 = "alzheimers_densenet169.pth"
69
+ model_201 = "alzheimers_densenet201.pth"
70
 
71
  # -----------------------------
72
  # SINGLE MODEL PREDICTION