Update app.py

making changes for robustness

app.py

@@ -338,10 +338,18 @@ def resize_img(img: np.ndarray, resize_dim):
 
 
 def predict(image, offset_inches):
-    drawer_img = yolo_detect(image, ["box"])
-    shrunked_img = make_square(shrink_bbox(drawer_img, 0.8))
+    try:
+        drawer_img = yolo_detect(image, ["box"])
+        shrunked_img = make_square(shrink_bbox(drawer_img, 0.8))
+    except:
+        raise gr.Error("Unable to DETECT DRAWER, please take another picture with different magnification level!")
+
     # Detect the scaling reference square
-    reference_obj_img, scaling_box_coords = detect_reference_square(shrunked_img)
+    try:
+        reference_obj_img, scaling_box_coords = detect_reference_square(shrunked_img)
+    except:
+        raise gr.Error("Unable to DETECT REFERENCE BOX, please take another picture with different magnification level!")
+
     # reference_obj_img_scaled = shrink_bbox(reference_obj_img, 1.2)
     # make the image sqaure so it does not effect the size of objects
     reference_obj_img = make_square(reference_obj_img)