Update app.py

app.py

@@ -7,11 +7,11 @@ from sam2.build_sam import build_sam2
 from sam2.sam2_image_predictor import SAM2ImagePredictor
 import gradio as gr
 
-hf_hub_download(repo_id = "merve/sam2-hiera-tiny", filename="sam2_hiera_tiny.pt", local_dir = "./")
+hf_hub_download(repo_id = "merve/sam2-hiera-small", filename="sam2_hiera_small.pt", local_dir = "./")
 
 DEVICE = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
-CHECKPOINT = f"./sam2_hiera_tiny.pt"
-CONFIG = "sam2_hiera_t.yaml"
+CHECKPOINT = f"./sam2_hiera_small.pt"
+CONFIG = "sam2_hiera_s.yaml"
 
 sam2_model = build_sam2(CONFIG, CHECKPOINT, device=DEVICE, apply_postprocessing=False)
 predictor = SAM2ImagePredictor(sam2_model)
@@ -56,7 +56,9 @@ demo = gr.Interface(
     title="OWLv2 🤝 SAMv2",
     description=description,
     examples=[
-        ["./cats.png", "cat", 0.1],
+        ["./bird.jpg", "bird", 0.15],
+        ["./buddha.JPG", "buddha", 0.15],
+        
     ],
     cache_examples=True
 )