Upload 3 files

README.md

@@ -1,12 +1,9 @@
----
-title: SAM Gardio
-emoji: 🐢
-colorFrom: yellow
-colorTo: purple
-sdk: gradio
-sdk_version: 5.23.3
-app_file: app.py
-pinned: false
----
+# SAM Gradio App for Hugging Face Space
+
+A simple Gradio demo for Meta's Segment Anything Model (SAM), deployable on Hugging Face Spaces.
+
+## Instructions
+1. Upload the `sam_vit_b.pth` model weights into the root of this repo on Hugging Face.
+2. Hugging Face will automatically install the dependencies and launch the app.
+3. Interact with the model via Gradio UI.
 
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,42 @@
+import gradio as gr
+from segment_anything import sam_model_registry, SamPredictor
+from PIL import Image
+import numpy as np
+import torch
+
+# Load model
+sam_checkpoint = "sam_vit_b.pth"  # Upload this manually to the Hugging Face Space
+model_type = "vit_b"
+device = "cuda" if torch.cuda.is_available() else "cpu"
+
+sam = sam_model_registry[model_type](checkpoint=sam_checkpoint).to(device)
+predictor = SamPredictor(sam)
+
+def segment_with_sam(image):
+    image_np = np.array(image.convert("RGB"))
+    predictor.set_image(image_np)
+
+    # Dummy point for prompt (center)
+    h, w, _ = image_np.shape
+    point = np.array([[w // 2, h // 2]])
+    label = np.array([1])
+
+    masks, scores, _ = predictor.predict(
+        point_coords=point,
+        point_labels=label,
+        multimask_output=False
+    )
+
+    mask = masks[0]
+    mask_img = (mask[..., None] * np.array([255, 0, 0])).astype(np.uint8)
+    overlay = Image.fromarray((0.5 * image_np + 0.5 * mask_img).astype(np.uint8))
+
+    return overlay
+
+iface = gr.Interface(fn=segment_with_sam, 
+                     inputs=gr.Image(type="pil"), 
+                     outputs=gr.Image(type="pil"),
+                     title="Segment Anything with SAM",
+                     description="Simple SAM demo using Gradio.")
+
+iface.launch()

requirements.txt

@@ -0,0 +1,7 @@
+gradio
+opencv-python
+numpy
+torch
+torchvision
+pillow
+git+https://github.com/facebookresearch/segment-anything.git