Update app.py

app.py

@@ -118,7 +118,7 @@ def exclude_scaling_box(
     bbox: np.ndarray,
     orig_size: tuple,
     processed_size: tuple,
-    expansion_factor: float = 1.5,
+    expansion_factor: float = 1.2,
 ) -> np.ndarray:
     # Unpack the bounding box
     x_min, y_min, x_max, y_max = map(int, bbox)
@@ -324,7 +324,7 @@ def resize_img(img: np.ndarray, resize_dim):
 def predict(image, offset_inches):
     try:
         drawer_img = yolo_detect(image, ["box"])
-        shrunked_img = make_square(shrink_bbox(drawer_img, 0.8))
+        shrunked_img = make_square(shrink_bbox(drawer_img, 0.85))
     except:
         raise gr.Error("Unable to DETECT DRAWER, please take another picture with different magnification level!")
 
@@ -374,7 +374,7 @@ def predict(image, offset_inches):
         scaling_box_coords,
         orig_size,
         processed_size,
-        expansion_factor=1.5,
+        expansion_factor=1.2,
     )
     objects_mask = resize_img(
         objects_mask, (shrunked_img.shape[1], shrunked_img.shape[0])