"git@developer.sourcefind.cn:chenpangpang/open-webui.git" did not exist on "e1d4c87ff07ff4a5c3ea4f94650098e6084a555e"
Commit 4dc092b7 authored by zhougaofeng's avatar zhougaofeng
Browse files

Update UNIPipe.py

parent 04d8ad48
......@@ -41,7 +41,7 @@ class UNIPipe(AbsPipe):
lang=self.lang, layout_model=self.layout_model,
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:
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,
......@@ -49,7 +49,7 @@ class UNIPipe(AbsPipe):
lang=self.lang, layout_model=self.layout_model,
formula_enable=self.formula_enable, table_enable=self.table_enable)
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,
start_page_id=self.start_page_id, end_page_id=self.end_page_id,
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