Update app.py

app.py

@@ -64,11 +64,9 @@ def load_model(path):
 # -----------------------------
 # LOAD ALL MODELS
 # -----------------------------
-BASE = Path("saved_models")
-
-model_121 = load_model(BASE / "alzheimers_densenet121.pth")
-model_169 = load_model(BASE / "alzheimers_densenet169.pth")
-model_201 = load_model(BASE / "alzheimers_densenet201.pth")
+model_121 = "alzheimers_densenet121.pth"
+model_169 = "alzheimers_densenet169.pth"
+model_201 = "alzheimers_densenet201.pth"
 
 # -----------------------------
 # SINGLE PREDICT FUNCTION