Update app.py

app.py

@@ -18,9 +18,6 @@ device = torch.device("cpu")
 save_dir = "/home/user/app/saved_predictions"
 os.makedirs(save_dir, exist_ok=True)
 
-# Placeholder image for invalid uploads
-invalid_img = Image.new("RGB", (224, 224), color=(200, 200, 200))
-
 # -----------------------
 # Load model
 # -----------------------
@@ -42,22 +39,6 @@ transform = transforms.Compose([
                          [0.229, 0.224, 0.225])
 ])
 
-# -----------------------
-# Helper: basic fundus check
-# -----------------------
-def looks_like_fundus(image):
-    """
-    Basic heuristic to check if an image is likely a retinal fundus scan.
-    Fundus images tend to have a bright circular region (retina)
-    with a darker background.
-    """
-    img = np.array(image.convert("L").resize((224, 224)))
-    thresh = img > 40  # bright region
-    white_ratio = np.sum(thresh) / img.size
-
-    # Fundus images usually occupy ~40–75% of the area
-    return 0.40 < white_ratio < 0.75
-
 # -----------------------
 # Predict and save
 # -----------------------