Update weights and app

main.py

@@ -94,12 +94,13 @@ def predict(image_input):
     # Filter out masks with low scores.
     mask_images = mask_images[scores > 0.5]
 
-    # For each mask, get the bounding box.
-    for mask in mask_images:
-        bbox_list.extend(get_bbox(mask.astype(np.uint8)))
+    if len(mask_images) == 0:
+        return []
 
-    # Return the bounding boxes.
-    return bbox_list
+    # Merge the masks into one mask.
+    mask_image = np.sum(mask_images, axis=0)    
+
+    return get_bbox(mask_image.astype(np.uint8))
 
 
 def run():

model/table-det/model.pth

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:4ed1343dc35386eda4f64af2f67413ef8b15455fc55c753b6222c7d5287b90cf
-size 351614497
+oid sha256:ff4f4b8809fb9c886e96cc539cae6517fbdc289406d8b3649918d3fa497c6286
+size 352081121