handle 0 points

app.py

@@ -65,6 +65,10 @@ def load_default_pipeline():
 @spaces.GPU()
 def predict_masks(image, points):
     """Predict a single mask from the image based on selected points."""
+    
+    if not points:
+        return image  # Return the original image if no points are selected
+
     PREDICTOR = SAM2ImagePredictor.from_pretrained(SAM_MODEL, device=DEVICE)
 
     image_np = np.array(image)