implemented SAM segment anything

app.py

@@ -1,8 +1,20 @@
 import gradio as gr
+from segment_anything import SamAutomaticMaskGenerator, sam_model_registry
+import supervision as sv
+
 
 
 def snap(image, video):
-    return [image, video]
+    MODEL_TYPE = "vit_b"
+    checkpoint = "checkpoints/sam_vit_b_01ec64.pth"
+    sam = sam_model_registry[MODEL_TYPE](checkpoint=checkpoint)
+    mask_generator = SamAutomaticMaskGenerator(sam)
+    #mask_generator = SamAutomaticMaskGenerator(sam, points_per_side=50)
+    sam_result = mask_generator.generate(image)
+    mask_annotator = sv.MaskAnnotator()
+    detections = sv.Detections.from_sam(sam_result=sam_result)
+    annotated_image = mask_annotator.annotate(scene=image.copy(), detections=detections)
+    return [annotated_image, video]
 
 
 demo = gr.Interface(

requirements.txt

@@ -1 +1,7 @@
 gradio
+huggingface_hub
+segment-anything
+supervision
+torch
+torchvision
+opencv-python