napatswift commited on
Commit
992ad70
1 Parent(s): f774bb9

Update app

Browse files
Files changed (1) hide show
  1. main.py +2 -2
main.py CHANGED
@@ -91,10 +91,10 @@ def predict(image_input):
91
  bbox_list = []
92
 
93
  # Filter out the masks with a score less than 0.5.
94
- filtered_mask_images = mask_images[scores > 0.5]
95
 
96
  # Get the bounding boxes of the tables.
97
- for mask_image in filtered_mask_images:
98
  bbox_list.extend(get_bbox(mask_image.astype(np.uint8)))
99
 
100
  return bbox_list
 
91
  bbox_list = []
92
 
93
  # Filter out the masks with a score less than 0.5.
94
+ filtered_mask_images = mask_images
95
 
96
  # Get the bounding boxes of the tables.
97
+ for mask_image in mask_images:
98
  bbox_list.extend(get_bbox(mask_image.astype(np.uint8)))
99
 
100
  return bbox_list