소스 검색

add table recognition success detect

liukaiwen 1 년 전
부모
커밋
377b49eb24
1개의 변경된 파일1개의 추가작업 그리고 2개의 파일을 삭제
  1. 1 2
      magic_pdf/model/pdf_extract_kit.py

+ 1 - 2
magic_pdf/model/pdf_extract_kit.py

@@ -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