Skip to content

GitLab

Unverified Commit 0d632833 authored by Xiaomeng Zhao's avatar Xiaomeng Zhao Committed by GitHub
Browse files

Merge pull request #1060 from myhloli/dev 

fix(table): add null check for OCR result in rapid table prediction
parents 958168b3 18aa1a20
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
magic_pdf/model/sub_modules/table/rapidtable/rapid_table.py
View file @ 0d632833
...@@ -10,5 +10,7 @@ class RapidTableModel(object): ...@@ -10,5 +10,7 @@ class RapidTableModel(object):
def predict(self, image): def predict(self, image):
ocr_result, _ = self.ocr_engine(np.asarray(image)) ocr_result, _ = self.ocr_engine(np.asarray(image))
if ocr_result is None:
return None, None, None
html_code, table_cell_bboxes, elapse = self.table_model(np.asarray(image), ocr_result) html_code, table_cell_bboxes, elapse = self.table_model(np.asarray(image), ocr_result)
return html_code, table_cell_bboxes, elapse return html_code, table_cell_bboxes, elapse
\ No newline at end of file
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