Update app.py

app.py

@@ -44,7 +44,10 @@ def segment_to_bbox(segment_indexs):
             if val == 1:
                 x_points.append(x)
                 y_points.append(y)
-    return [np.min(x_points), np.min(y_points), np.max(x_points), np.max(y_points)]
+    if x_points and y_points:
+        return [np.min(x_points), np.min(y_points), np.max(x_points), np.max(y_points)]
+    else:
+        return [0,0,0,0]
 
 def clipseg_prediction(image):
 
@@ -74,13 +77,13 @@ def clipseg_prediction(image):
 
     # Vehicle checking
     if bbox_area(vehicle_bbox) > bbox_area(damage_bbox):
-        return True, bbox_normalization(damage_bbox, img_w, img_h)
+        return True, [bbox_normalization(damage_bbox, img_w, img_h)]
     else:
         return False, [[]]
      
 
 @torch.no_grad()
-def foward_pass(image_input: np.ndarray, points: List[int]) -> np.ndarray:
+def foward_pass(image_input: np.ndarray, points: List[List[int]]) -> np.ndarray:
     print('SAM-Segmentation-started------->')
     global cache_data
     image_input = Image.fromarray(image_input)