Update app.py

app.py

@@ -26,22 +26,22 @@ def infer(img):
 
   # get point prompt
   img_arr = np.array(point_prompt)
-  nonzero_indices = np.nonzero(img_arr)
-  if nonzero_indices is None:
-      gr.Error("Please select a point.")
-  center_x = int(np.mean(nonzero_indices[1]))
-  center_y = int(np.mean(nonzero_indices[0]))
-  input_point = np.array([[center_x, center_y]])
-
-  input_label = np.array([1])
-  masks, scores, logits = predictor.predict(
+  if not np.any(img_arr):
+    gr.Error("Please select a point on top of the image.")
+  else:
+    nonzero_indices = np.nonzero(img_arr)
+    img_arr = np.array(point_prompt)
+    nonzero_indices = np.nonzero(img_arr)
+    center_x = int(np.mean(nonzero_indices[1]))
+    center_y = int(np.mean(nonzero_indices[0]))
+    input_point = np.array([[center_x, center_y]])
+    input_label = np.array([1])
+    masks, scores, logits = predictor.predict(
       point_coords=input_point,
       point_labels=input_label,
   )
-
-
-  result_label = [(masks[0, :, :], "mask")]
-  return image, result_label
+    result_label = [(masks[0, :, :], "mask")]
+    return image, result_label
 
 
 with gr.Blocks() as demo:
@@ -53,8 +53,7 @@ with gr.Blocks() as demo:
             im = gr.ImageEditor(
                 type="pil"
             )
-            submit_btn = gr.Button()
         output = gr.AnnotatedImage()
-    submit_btn.click(infer, inputs=im, outputs=output)
+    im.change(infer, inputs=im, outputs=output)
 
 demo.launch(debug=True)