Update app.py

app.py

@@ -34,12 +34,11 @@ def download_models():
     stroke_model = load_model("cnn_stroke_model.keras")
 
 
-pain_model = resnet18(weights=None)
-pain_model.fc = nn.Linear(pain_model.fc.in_features, 1)
-pain_model.load_state_dict(torch.load("pain_model.pth", map_location=torch.device("cpu")))
-pain_model.eval()
+    pain_model = resnet18(weights=None)
+    pain_model.fc = nn.Linear(pain_model.fc.in_features, 1)
+    pain_model.load_state_dict(torch.load("pain_model.pth", map_location=torch.device("cpu")))
+    pain_model.eval()
 
-    st.write("✅ Models loaded.")
 
     return stroke_model, pain_model