Parcourir la source

refactor(model): update batch analyze logic for rapid table model

- 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
myhloli il y a 10 mois
Parent
commit
452a9c0bcd
1 fichiers modifiés avec 1 ajouts et 1 suppressions
  1. 1 1
      magic_pdf/model/batch_analyze.py

+ 1 - 1
magic_pdf/model/batch_analyze.py

@@ -161,7 +161,7 @@ class BatchAnalyze:
                     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