Update app.py

app.py

@@ -37,8 +37,8 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
 # we initialize model and processor
 checkpoint = "google/owlv2-base-patch16-ensemble"
 detector = pipeline(model=checkpoint, task="zero-shot-object-detection", device=device)
-sam_model = SamModel.from_pretrained("jadechoghari/robustsam-vit-base").to(device)
-sam_processor = SamProcessor.from_pretrained("jadechoghari/robustsam-vit-base")
+sam_model = SamModel.from_pretrained("jadechoghari/robustsam-vit-huge").to(device)
+sam_processor = SamProcessor.from_pretrained("jadechoghari/robustsam-vit-huge")
 
 def apply_mask(image, mask, color):
     """Apply a mask to an image with a specific color."""