Update app and weights

main.py

@@ -86,26 +86,28 @@ def get_bbox(mask_array):
 
 def predict(image_input):
     logger.info(f"Image input: {image_input}")
-    
+
     # Inference the tables in the image.
     result = inference_detector(table_det, image_input)
 
     # Get the masks of the tables.
     mask_images = result.pred_instances.masks.cpu().numpy()
     scores = result.pred_instances.scores.cpu().numpy()
+    bboxes = result.pred_instances.bboxes.cpu().numpy()
     
     logger.info(f"Result: {result}")
 
     bbox_list = []
 
     # Filter out the masks with a score less than 0.5.
-    filtered_mask_images = mask_images
+    filtered_mask_images = mask_images[scores > 0.5]
+    filtered_bboxes = bboxes[scores > 0.5]
 
     # Get the bounding boxes of the tables.
-    for mask_image in mask_images:
+    for mask_image in filtered_mask_images:
         bbox_list.extend(get_bbox(mask_image.astype(np.uint8)))
 
-    return bbox_list
+    return {'rect-fit': bbox_list, 'bbox': filtered_bboxes}
 
 
 def run():

model/table-det/model.pth

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:e84722e31515bf2415bec7fafbb3f2d9ebbf058e7003b91d798e4cdb9219a58e
-size 351647241
+oid sha256:fc857b4c227cf5ee04c66fc6909cb4d63dd19f412cec20721e1d79b0b9fb242e
+size 351847113