瀏覽代碼

fix UNIPipe cli

赵小蒙 1 年之前
父節點
當前提交
84bc6520b6
共有 1 個文件被更改,包括 5 次插入1 次删除
  1. 5 1
      magic_pdf/cli/magicpdf.py

+ 5 - 1
magic_pdf/cli/magicpdf.py

@@ -60,7 +60,11 @@ def prepare_env(pdf_file_name, method):
 
 def _do_parse(pdf_file_name, pdf_bytes, model_list, parse_method, image_writer, md_writer, image_dir, local_md_dir):
     if parse_method == "auto":
-        pipe = UNIPipe(pdf_bytes, model_list, image_writer, is_debug=True)
+        jso_useful_key = {
+            "_pdf_type": "",
+            "model_list": model_list
+        }
+        pipe = UNIPipe(pdf_bytes, jso_useful_key, image_writer, is_debug=True)
     elif parse_method == "txt":
         pipe = TXTPipe(pdf_bytes, model_list, image_writer, is_debug=True)
     elif parse_method == "ocr":