Skip to content

GitLab

Commit 4dc092b7 authored by zhougaofeng's avatar zhougaofeng
Browse files

Update UNIPipe.py

parent 04d8ad48
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
magic_pdf/pipe/UNIPipe.py
View file @ 4dc092b7
...@@ -41,7 +41,7 @@ class UNIPipe(AbsPipe): ...@@ -41,7 +41,7 @@ class UNIPipe(AbsPipe):
lang=self.lang, layout_model=self.layout_model, lang=self.lang, layout_model=self.layout_model,
formula_enable=self.formula_enable, table_enable=self.table_enable) formula_enable=self.formula_enable, table_enable=self.table_enable)
def pipe_parse(self,ocr_status,config_path,local_image_dir): def pipe_parse(self,config_path,local_image_dir):
if self.pdf_type == self.PIP_TXT: if self.pdf_type == self.PIP_TXT:
self.pdf_mid_data = parse_union_pdf(self.pdf_bytes, self.model_list, self.image_writer, self.pdf_mid_data = parse_union_pdf(self.pdf_bytes, self.model_list, self.image_writer,
is_debug=self.is_debug, input_model_is_empty=self.input_model_is_empty, is_debug=self.is_debug, input_model_is_empty=self.input_model_is_empty,
...@@ -49,7 +49,7 @@ class UNIPipe(AbsPipe): ...@@ -49,7 +49,7 @@ class UNIPipe(AbsPipe):
lang=self.lang, layout_model=self.layout_model, lang=self.lang, layout_model=self.layout_model,
formula_enable=self.formula_enable, table_enable=self.table_enable) formula_enable=self.formula_enable, table_enable=self.table_enable)
elif self.pdf_type == self.PIP_OCR: elif self.pdf_type == self.PIP_OCR:
self.pdf_mid_data = parse_ocr_pdf(ocr_status,config_path,local_image_dir,self.pdf_bytes, self.model_list, self.image_writer, self.pdf_mid_data = parse_ocr_pdf(config_path,local_image_dir,self.pdf_bytes, self.model_list, self.image_writer,
is_debug=self.is_debug, is_debug=self.is_debug,
start_page_id=self.start_page_id, end_page_id=self.end_page_id, start_page_id=self.start_page_id, end_page_id=self.end_page_id,
lang=self.lang) lang=self.lang)
......
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