pred

app.py

@@ -83,11 +83,11 @@ def predict_masks(prompts):
 
     DEVICE = torch.device("cuda")
     SAM_MODEL = "facebook/sam2.1-hiera-large"
-    if PREDICTOR is None:
-        PREDICTOR = SAM2ImagePredictor.from_pretrained(SAM_MODEL, device=DEVICE)
-    else:
-        PREDICTOR = PREDICTOR
-
+    # if PREDICTOR is None:
+    #     PREDICTOR = SAM2ImagePredictor.from_pretrained(SAM_MODEL, device=DEVICE)
+    # else:
+    #     PREDICTOR = PREDICTOR
+    PREDICTOR = SAM2ImagePredictor.from_pretrained(SAM_MODEL, device=DEVICE)
 
     """Predict a single mask from the image based on selected points."""
     image = np.array(prompts["image"])  # Convert the image to a numpy array