xiaoyao9184 commited on
Commit
41cfc73
·
verified ·
1 Parent(s): 38e9235

Synced repo using 'sync_with_huggingface' Github Action

Browse files
Files changed (2) hide show
  1. gradio_app.py +1 -1
  2. 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 == "Table"]
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.2
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