Added samples

.gitattributes

@@ -14,6 +14,11 @@
 *.npy filter=lfs diff=lfs merge=lfs -text
 *.npz filter=lfs diff=lfs merge=lfs -text
 *.onnx filter=lfs diff=lfs merge=lfs -text
+*.mp4 filter=lfs diff=lfs merge=lfs -text
+*.jpg filter=lfs diff=lfs merge=lfs -text
+*.gif filter=lfs diff=lfs merge=lfs -text
+*.png filter=lfs diff=lfs merge=lfs -text
+*.webp filter=lfs diff=lfs merge=lfs -te
 *.ot filter=lfs diff=lfs merge=lfs -text
 *.parquet filter=lfs diff=lfs merge=lfs -text
 *.pb filter=lfs diff=lfs merge=lfs -text

app.py

@@ -78,24 +78,61 @@ def clear_points():
     state["labels"].clear()
     return state["original_image_rgb"], None
 
+def update_image_state(new_image):
+    """
+    Called only when the user actually loads or picks a new image.
+    Reset everything here.
+    """
+    if new_image is None or np.array_equal(new_image, state["original_image_rgb"]):
+        return None, None
+
+    bgr = cv.cvtColor(new_image, cv.COLOR_RGB2BGR)
+    state["image"] = bgr
+    state["original_image_rgb"] = new_image.copy()
+
+    state["points"].clear()
+    state["labels"].clear()
+
+    return None, None
+
 # Gradio UI
-with gr.Blocks() as demo:
-    gr.Markdown("## EfficientSAM: Interactive Image Segmentation")
-    gr.Markdown("Click to add **foreground points** (max 6), then click **Segment**. Use **Clear All** to reset everything or **Clear Points** to just remove dots.")
+with gr.Blocks(css='''.example * {
+    font-style: italic;
+    font-size: 18px !important;
+    color: #0ea5e9 !important;
+    }''') as demo:
+
+    gr.Markdown("### EfficientSAM: Interactive Image Segmentation")
+    gr.Markdown("Click to add up to 6 **foreground points**, then click **Segment**. Use **Clear All** to reset, or **Clear Points** to remove only clicks.")
 
     with gr.Row():
-        input_img = gr.Image(type="numpy", label="Input Image", interactive=True)
-        output_img = gr.Image(type="numpy", label="Segmentation Result")
+        input_image = gr.Image(type="numpy", label="Upload Image", interactive=True)
+        prompt_overlay = gr.Image(type="numpy", label="Your Clicks", interactive=False)
+        output_image = gr.Image(type="numpy", label="Segmentation Result")
+
+    input_image.change(fn=update_image_state, inputs=input_image, outputs=[output_image, prompt_overlay])
 
     with gr.Row():
         segment_btn = gr.Button("Segment", variant="primary")
         clear_points_btn = gr.Button("Clear Points")
         clear_all_btn = gr.Button("Clear All")
 
-    input_img.select(fn=segment_image_with_prompts, inputs=input_img, outputs=input_img)
-    segment_btn.click(fn=run_segmentation, outputs=output_img)
-    clear_points_btn.click(fn=clear_points, outputs=[input_img, output_img])
-    clear_all_btn.click(fn=clear_all, outputs=[input_img, output_img])
+    segment_btn.click(fn=run_segmentation, outputs=output_image)
+    clear_points_btn.click(fn=clear_points, outputs=[input_image, output_image])
+    clear_all_btn.click(fn=clear_all, outputs=[input_image, output_image])
+    input_image.select(fn=segment_image_with_prompts, inputs=input_image, outputs=prompt_overlay)
+
+    gr.Markdown("Click on any example to try it.", elem_classes=["example"])
+
+    gr.Examples(
+        examples=[
+            ["examples/dog416.png"],
+            ["examples/messi5.jpg"],
+            ["examples/selfie.jpg"],
+            ["examples/space_shuttle.jpg"]
+        ],
+        inputs=input_image
+    )
 
 if __name__ == "__main__":
     demo.launch()