Fix Space startup and inference robustness

app.py

@@ -66,16 +66,16 @@ def sam_point_inference(image, x, y, *, slim=False):
 
 def infer_point(img):
     if img is None:
-        gr.Error("Please upload an image and select a point.")
+        raise 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.")
+        raise 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]
     total_image = img["composite"]
     img_arr = np.array(point_prompt)
     if not np.any(img_arr):
-        gr.Error("Please select a point on top of the image.")
+        raise gr.Error("Please select a point on top of the image.")
     else:
         nonzero_indices = np.nonzero(img_arr)
         img_arr = np.array(point_prompt)
@@ -90,10 +90,10 @@ def infer_box(prompts):
     # background (original image) layers[0] ( point prompt) composite (total image)
     image = prompts["image"]
     if image is None:
-      gr.Error("Please upload an image and draw a box before submitting")
+      raise gr.Error("Please upload an image and draw a box before submitting")
     points = prompts["points"][0]
     if points is None:
-      gr.Error("Please draw a box before submitting.")
+      raise gr.Error("Please draw a box before submitting.")
     print(points)
 
     # x_min = points[0] x_max = points[3] y_min = points[1] y_max = points[4]