Update app.py

app.py

@@ -47,6 +47,8 @@ def segment_to_bbox(segment_indexs):
     return [np.min(x_points), np.min(y_points), np.max(x_points), np.max(y_points)]
 
 def clipseg_prediction(image):
+    img_w = image.width
+    img_h = image.height
     inputs = clip_processor(text=prompts, images=[image] * len(prompts), padding="max_length", return_tensors="pt")
     # predict
     with torch.no_grad():
@@ -71,7 +73,7 @@ def clipseg_prediction(image):
 
     # Vehicle checking
     if bbox_area(vehicle_bbox) > bbox_area(damage_bbox):
-        return True, [bbox_normalization(damage_bbox)]
+        return True, [bbox_normalization(damage_bbox, img_w, img_h)]
     else:
         return False, [[]]