update to 2.1

app.py

@@ -13,7 +13,6 @@ from sam2_mask import create_sam2_mask_interface
 
 #from sam2.sam2_image_predictor import SAM2ImagePredictor
 
-sam2_mask_tab = create_sam2_mask_interface()
 
 MODELS = {
     "RealVisXL V5.0 Lightning": "SG161222/RealVisXL_V5.0_Lightning",
@@ -315,6 +314,8 @@ def clear_cache():
     torch.cuda.empty_cache()
     return gr.update(value="Cache cleared!")
 
+sam2_mask_tab = create_sam2_mask_interface()
+
 css = """
 .nulgradio-container {
     width: 86vw !important;

sam2_mask.py

@@ -8,7 +8,7 @@ import numpy as np
 from PIL import Image as PILImage
 
 # Initialize SAM2 predictor
-MODEL = "facebook/sam2-hiera-large"
+MODEL = "facebook/sam2.1-hiera-large"
 DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 PREDICTOR = SAM2ImagePredictor.from_pretrained(MODEL, device=DEVICE)