Commit c19bb36e authored by zhougaofeng's avatar zhougaofeng
Browse files

Update pdf_parse_by_ocr.py

parent 23eff75f
......@@ -3,7 +3,7 @@ from magic_pdf.data.dataset import PymuDocDataset
from magic_pdf.pdf_parse_union_core_v2 import pdf_parse_union
def parse_pdf_by_ocr(ocr_status,config_path,local_image_dir,pdf_bytes,
def parse_pdf_by_ocr(config_path,local_image_dir,pdf_bytes,
model_list,
imageWriter,
start_page_id=0,
......@@ -11,7 +11,7 @@ def parse_pdf_by_ocr(ocr_status,config_path,local_image_dir,pdf_bytes,
debug_mode=False,
):
dataset = PymuDocDataset(pdf_bytes)
return pdf_parse_union(ocr_status,config_path,local_image_dir,dataset,
return pdf_parse_union(config_path,local_image_dir,dataset,
model_list,
imageWriter,
SupportedPdfParseMethod.OCR,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment