Skip to content

Commit

Permalink
refactor(model): update batch analyze logic for rapid table model
Browse files Browse the repository at this point in the history
- Modify the batch analyze process to handle the rapid table model's output
- Add logic_points variable to capture additional output from rapid table prediction
  • Loading branch information
myhloli committed Jan 16, 2025
1 parent 61f75b3 commit 452a9c0
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion magic_pdf/model/batch_analyze.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit 452a9c0

Please sign in to comment.