Update app.py

app.py

@@ -105,6 +105,17 @@ with gr.Blocks() as demo:
 
     image_input = gr.Image(type="pil", label="📷 Upload Image (for ViT)", visible=True)
 
+    example_image = {"🔵 Benign Example Image": "images/benign (1)_aug_0.png",
+                     "🔴 Malignant Example Image": "images/malignant (1)_aug_0.png"}
+
+    with gr.Row(visible=True) as example_image_row:
+      example_buttons = []
+    for label, path in example_images.items():
+        with gr.Column():
+            gr.Image(value=path, label=label, interactive=False, height=100)
+            btn = gr.Button(f"Use {label.split()[1]}")
+            example_buttons.append((btn, path))
+    
     feature_inputs = [gr.Number(label=feature) for feature in feature_names]
 
     # Organizing feature inputs into rows of 3 columns