diff --git a/magic_pdf/model/batch_analyze.py b/magic_pdf/model/batch_analyze.py index 630832ce..496bb43e 100644 --- a/magic_pdf/model/batch_analyze.py +++ b/magic_pdf/model/batch_analyze.py @@ -161,7 +161,7 @@ def __call__(self, images: list) -> list: elif self.model.table_model_name == MODEL_NAME.TABLE_MASTER: html_code = self.model.table_model.img2html(new_image) elif self.model.table_model_name == MODEL_NAME.RAPID_TABLE: - html_code, table_cell_bboxes, elapse = ( + html_code, table_cell_bboxes, logic_points, elapse = ( self.model.table_model.predict(new_image) ) run_time = time.time() - single_table_start_time