Update app.py

app.py

@@ -40,6 +40,10 @@ transform = transforms.Compose([
 
 # Predict and save
 def predict_retinopathy(image):
+    # Validate image
+    if not looks_like_fundus(image):
+        return None, "❌ Please upload a valid retinal (fundus) scan."
+
     timestamp = datetime.datetime.now().strftime("%Y%m%d_%H%M%S")
     img = image.convert("RGB").resize((224, 224))
     img_tensor = transform(img).unsqueeze(0).to(device)