Fix zero point error

app.py

@@ -1,7 +1,8 @@
+import os
+
 import gradio as gr
 import numpy as np
 import torch
-import os
 from mobile_sam import SamAutomaticMaskGenerator, SamPredictor, sam_model_registry
 from PIL import ImageDraw
 from utils.tools import box_prompt, format_results, point_prompt
@@ -109,9 +110,18 @@ def segment_with_points(
     new_h = int(h * scale)
     image = image.resize((new_w, new_h))
 
-    scaled_points = np.array([[int(x * scale) for x in point] for point in global_points])
+    scaled_points = np.array(
+        [[int(x * scale) for x in point] for point in global_points]
+    )
     scaled_point_label = np.array(global_point_label)
 
+    if scaled_points.size == 0 and scaled_point_label.size == 0:
+        print("No points selected")
+        return image, image
+
+    print(scaled_points, scaled_points is not None)
+    print(scaled_point_label, scaled_point_label is not None)
+
     nd_image = np.array(image)
     predictor.set_image(nd_image)
     masks, scores, logits = predictor.predict(