Update app and weights

main.py

@@ -84,21 +84,16 @@ def predict(image_input):
     # Inference the tables in the image.
     result = inference_detector(table_det, image_input)
 
-    # Create a list to store the bounding boxes.
-    bbox_list = []
-
     # Get the masks of the tables.
     mask_images = result.pred_instances.masks.cpu().numpy()
     scores = result.pred_instances.scores.cpu().numpy()
 
-    # Filter out masks with low scores.
-    mask_images = mask_images[scores > 0.5]
-
+    # If there are no tables, return an empty list.
     if len(mask_images) == 0:
         return []
 
-    # Merge the masks into one mask.
-    mask_image = np.sum(mask_images, axis=0)    
+    # Get mask with highest score
+    mask_image = mask_images[scores.argmax()]
 
     return get_bbox(mask_image.astype(np.uint8))
 

model/table-det/model.pth

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:ff4f4b8809fb9c886e96cc539cae6517fbdc289406d8b3649918d3fa497c6286
-size 352081121
+oid sha256:5e04ce5891ea3c2675b2b4b35268663b85f8a732c43844d912209c68cc4903d3
+size 351141001