Filter predictions

main.py

@@ -89,6 +89,10 @@ def predict(image_input):
 
     # 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]
 
     # For each mask, get the bounding box.
     for mask in mask_images: