|
|
@@ -9,9 +9,6 @@ from magic_pdf.model.sub_modules.layout.layoutlmv3.model_init import Layoutlmv3_
|
|
|
from magic_pdf.model.sub_modules.mfd.yolov8.YOLOv8 import YOLOv8MFDModel
|
|
|
from magic_pdf.model.sub_modules.mfr.unimernet.Unimernet import UnimernetModel
|
|
|
|
|
|
-from magic_pdf.model.sub_modules.table.structeqtable.struct_eqtable import StructTableModel
|
|
|
-from magic_pdf.model.sub_modules.table.tablemaster.tablemaster_paddle import TableMasterPaddleModel
|
|
|
-
|
|
|
try:
|
|
|
from magic_pdf_ascend_plugin.model_plugin.ocr.paddleocr.ppocr_273_npu import ModifiedPaddleOCR
|
|
|
from magic_pdf_ascend_plugin.model_plugin.table.rapidtable.rapid_table_npu import RapidTableModel
|
|
|
@@ -21,6 +18,8 @@ except ImportError:
|
|
|
# from magic_pdf.model.sub_modules.ocr.paddleocr.ppocr_291_mod import ModifiedPaddleOCR
|
|
|
from magic_pdf.model.sub_modules.table.rapidtable.rapid_table import RapidTableModel
|
|
|
|
|
|
+from magic_pdf.model.sub_modules.table.structeqtable.struct_eqtable import StructTableModel
|
|
|
+from magic_pdf.model.sub_modules.table.tablemaster.tablemaster_paddle import TableMasterPaddleModel
|
|
|
|
|
|
def table_model_init(table_model_type, model_path, max_time, _device_='cpu', ocr_engine=None, table_sub_model_name=None):
|
|
|
if table_model_type == MODEL_NAME.STRUCT_EQTABLE:
|