Spaces:
Runtime error
Runtime error
napatswift
commited on
Commit
·
992ad70
1
Parent(s):
f774bb9
Update app
Browse files
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
|
| 95 |
|
| 96 |
# Get the bounding boxes of the tables.
|
| 97 |
-
for mask_image in
|
| 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
|