Update app.py

app.py

@@ -14,6 +14,10 @@ sam = sam_model_registry[model_type](checkpoint="./tinysam/tinysam.pth")
 predictor = SamPredictor(sam)
 
 def infer(img):
+  if img is None:
+      gr.Error("Please upload an image and select a point.")
+  if img["background"] is None:
+      gr.Error("Please upload an image and select a point.")
   # background (original image) layers[0] ( point prompt) composite (total image)
   image = img["background"].convert("RGB")
   point_prompt = img["layers"][0]
@@ -23,6 +27,8 @@ 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]])