Update app.py

app.py

@@ -28,8 +28,8 @@ vehicle_threshold = 0.5
 def bbox_normalization(bbox, width, height):
     height_coeff = height/352
     width_coeff = width/352
-    normalized_bbox = [int(bbox[0]*width_coeff), int(bbox[1]*height_coeff),
-                       int(bbox[2]*width_coeff), int(bbox[3]*height_coeff)]
+    normalized_bbox = [bbox[0]*width_coeff, bbox[1]*height_coeff,
+                       bbox[2]*width_coeff, bbox[3]*height_coeff]
     return normalized_bbox
 
 def bbox_area(bbox):
@@ -47,7 +47,7 @@ def segment_to_bbox(segment_indexs):
     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]
+        return [0.0,0.0,0.0,0.0]
 
 def clipseg_prediction(image):
 
@@ -77,9 +77,9 @@ 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, []
+        return False, [[]]
      
 
 @torch.no_grad()