Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -62,11 +62,11 @@ def load_model(path):
|
|
| 62 |
# -----------------------------
|
| 63 |
# MODEL PATHS
|
| 64 |
# -----------------------------
|
| 65 |
-
BASE = Path("saved_models")
|
| 66 |
|
| 67 |
-
|
| 68 |
-
|
| 69 |
-
|
|
|
|
| 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
|