Update app.py

app.py

@@ -87,6 +87,23 @@ def segment_image(pil_img):
                 2,
             )
 
+            def get_sample_images():
+    """
+    Get list of sample images from Eye_Dataset folder.
+    Returns a list of image file paths.
+    """
+    dataset_dir = "Eye_Dataset"
+    if not os.path.exists(dataset_dir):
+        return []
+    
+    sample_images = []
+    for filename in os.listdir(dataset_dir):
+        if filename.lower().endswith(('.png', '.jpg', '.jpeg')):
+            filepath = os.path.join(dataset_dir, filename)
+            sample_images.append(filepath)
+    
+    return sample_images[:10]  # Return max 10 sample images
+
     return image_np  # RGB numpy array works with gr.Image
 
 with gr.Blocks() as demo: