change sam checkpoint

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-small", filename="sam2_hiera_small.pt", local_dir = "./")
+hf_hub_download(repo_id = "merve/sam2-hiera-base-plus", filename="sam2_hiera_base_plus.pt", local_dir = "./")
 
 DEVICE = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
-CHECKPOINT = f"./sam2_hiera_small.pt"
-CONFIG = "sam2_hiera_s.yaml"
+CHECKPOINT = f"./sam2_hiera_base_plus.pt"
+CONFIG = "sam2_hiera_b+.yaml"
 
 sam2_model = build_sam2(CONFIG, CHECKPOINT, device=DEVICE, apply_postprocessing=False)
 predictor = SAM2ImagePredictor(sam2_model)