From 4e7511fb866216c48c43755fbcf080ecc1139262 Mon Sep 17 00:00:00 2001 From: icecraft Date: Wed, 11 Dec 2024 16:42:16 +0800 Subject: [PATCH] fix: dup classify pdf type --- magic_pdf/tools/common.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/magic_pdf/tools/common.py b/magic_pdf/tools/common.py index 4f416f40..d64328c4 100644 --- a/magic_pdf/tools/common.py +++ b/magic_pdf/tools/common.py @@ -123,6 +123,9 @@ def do_parse( formula_enable=formula_enable, table_enable=table_enable, ) + pipe_result = infer_result.pipe_txt_mode( + image_writer, debug_mode=True, lang=lang + ) else: infer_result = ds.apply( doc_analyze, @@ -132,9 +135,9 @@ def do_parse( formula_enable=formula_enable, table_enable=table_enable, ) - pipe_result = infer_result.pipe_auto_mode( - image_writer, debug_mode=True, lang=lang - ) + pipe_result = infer_result.pipe_ocr_mode( + image_writer, debug_mode=True, lang=lang + ) elif parse_method == 'txt': infer_result = ds.apply(