Synced repo using 'sync_with_huggingface' Github Action
Browse files- gradio_app.py +1 -1
- requirements.txt +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:
|
requirements.txt
CHANGED
@@ -1,5 +1,5 @@
|
|
1 |
torch==2.5.1
|
2 |
-
surya-ocr==0.9.
|
3 |
gradio==5.8.0
|
4 |
huggingface-hub==0.26.3
|
5 |
# gradio app need pdftext for run_ocr_errors
|
|
|
1 |
torch==2.5.1
|
2 |
+
surya-ocr==0.9.3
|
3 |
gradio==5.8.0
|
4 |
huggingface-hub==0.26.3
|
5 |
# gradio app need pdftext for run_ocr_errors
|