opensource/MinerU
3
0
Fork 0
mirror of https://github.com/opendatalab/MinerU.git synced 2026-09-24 23:10:23 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #354 from papayalove/master

Browse Source 
feat: add table recognition success detect
This commit is contained in:
Xiaomeng Zhao
2024-08-07 16:47:29 +08:00
committed by GitHub
parent 8da5328f18 fbf8f89be5
commit d93ea5b94f
1 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
magic_pdf/model/pdf_extract_kit.py
+1 -2
View File
@@ -36,7 +36,7 @@ from magic_pdf.model.pek_sub_modules.self_modify import ModifiedPaddleOCR
from magic_pdf.model.pek_sub_modules.structeqtable.StructTableModel import StructTableModel
def table_model_init(model_path, max_time=400, _device_='cpu'):
def table_model_init(model_path, max_time, _device_='cpu'):
table_model = StructTableModel(model_path, max_time=max_time, device=_device_)
return table_model
@@ -297,6 +297,5 @@ class CustomPEKModel:
if latex_code and latex_code.strip().endswith('end{tabular}'):
layout["latex"] = latex_code
else:
print(latex_code)
logger.warning(f"------------table recognition processing fails----------")
return layout_res