Synced repo using 'sync_with_huggingface' Github Action
Browse files- gradio_app.py +1 -1
gradio_app.py
CHANGED
|
@@ -84,7 +84,7 @@ def table_recognition(img, highres_img, skip_table_detection: bool) -> (Image.Im
|
|
| 84 |
table_imgs = [highres_img]
|
| 85 |
else:
|
| 86 |
_, layout_pred = layout_detection(img)
|
| 87 |
-
layout_tables_lowres = [l.bbox for l in layout_pred.bboxes if l.label
|
| 88 |
table_imgs = []
|
| 89 |
layout_tables = []
|
| 90 |
for tb in layout_tables_lowres:
|
|
|
|
| 84 |
table_imgs = [highres_img]
|
| 85 |
else:
|
| 86 |
_, layout_pred = layout_detection(img)
|
| 87 |
+
layout_tables_lowres = [l.bbox for l in layout_pred.bboxes if l.label in ["Table", "TableOfContents"]]
|
| 88 |
table_imgs = []
|
| 89 |
layout_tables = []
|
| 90 |
for tb in layout_tables_lowres:
|