Skip to content

GitLab

Commit 8b230796 authored by zhougaofeng's avatar zhougaofeng
Browse files

Update UNIPipe.py

parent 5513a575
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
magic_pdf/pipe/UNIPipe.py
View file @ 8b230796
......@@ -29,12 +29,12 @@ class UNIPipe(AbsPipe):
# logger.info('暂时强制使用ocr pdf')
self.pdf_type = self.PIP_OCR
def pipe_analyze(self):
def pipe_analyze(self,model):
if self.pdf_type == self.PIP_TXT:
self.model_list = doc_analyze(self.pdf_bytes, ocr=True,
self.model_list = doc_analyze(model,self.pdf_bytes, ocr=True,
start_page_id=self.start_page_id, end_page_id=self.end_page_id)
elif self.pdf_type == self.PIP_OCR:
self.model_list = doc_analyze(self.pdf_bytes, ocr=True,
self.model_list = doc_analyze(model,self.pdf_bytes, ocr=True,
start_page_id=self.start_page_id, end_page_id=self.end_page_id)
def pipe_parse(self):
......
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