gaotingquan 1 year ago
parent
commit
8a4726d7b6

+ 1 - 0
paddlex/inference/pipelines/table_recognition/table_recognition.py

@@ -46,6 +46,7 @@ class TableRecPipeline(BasePipeline):
         self.text_det_batch_size = text_det_batch_size
         self.text_rec_batch_size = text_rec_batch_size
         self.table_batch_size = table_batch_size
+        self.predictor_kwargs = predictor_kwargs
         super().__init__(predictor_kwargs=predictor_kwargs)
         self._build_predictor()
         # self.set_predictor(layout_batch_size, text_det_batch_size,text_rec_batch_size, table_batch_size)