update pp-chatocrv3 pipeline

paddlex/inference/pipelines/ppchatocrv3/ppchatocrv3.py

@@ -130,6 +130,8 @@ class PPChatOCRPipeline(TableRecPipeline):
             self.curve_pipeline = OCRPipeline(
                 text_det_model=self.curve_model,
                 text_rec_model=self.text_rec_model,
+                text_det_batch_size=self.text_det_batch_size,
+                text_rec_batch_size=self.text_rec_batch_size,
                 predictor_kwargs=self.predictor_kwargs,
             )
         else:

paddlex/inference/pipelines/table_recognition/table_recognition.py

@@ -55,9 +55,11 @@ class TableRecPipeline(BasePipeline):
     ):
         self.layout_predictor = self._create_model(model=self.layout_model)
         self.ocr_pipeline = OCRPipeline(
-            self.text_det_model,
-            self.text_rec_model,
-            self.predictor_kwargs,
+            text_det_model=self.text_det_model,
+            text_rec_model=self.text_rec_model,
+            text_det_batch_size=self.text_det_batch_size,
+            text_rec_batch_size=self.text_rec_batch_size,
+            predictor_kwargs=self.predictor_kwargs,
         )
         self.table_predictor = self._create_model(model=self.table_model)
         self._crop_by_boxes = CropByBoxes()

paddlex/pipelines/chatocrv3.yaml

@@ -0,0 +1,33 @@
+Global:
+  pipeline_name: chatocrv3
+  input: https://paddle-model-ecology.bj.bcebos.com/paddlex/imgs/demo_image/table_recognition.jpg
+  
+######################################## Setting ########################################
+# Please select the model from bellow `Support`
+
+Pipeline:
+  layout_model: PicoDet_layout_1x
+  table_model: SLANet
+  text_det_model: PP-OCRv4_mobile_det
+  text_rec_model: PP-OCRv4_mobile_rec
+  uvdoc_model: UVDoc
+  curve_model: PP-OCRv4_mobile_seal_det
+  oricls_model: PP-LCNet_x1_0_doc_ori
+  llm_name: "ernie-3.5"
+  llm_params: 
+    api_type: qianfan
+    ak: 
+    sk: 
+  task_prompt_yaml: None
+  user_prompt_yaml:
+  layout_batch_size: 1
+  text_det_batch_size: 1
+  text_rec_batch_size: 1
+  table_batch_size: 1
+  uvdoc_batch_size: 1
+  curve_batch_size: 1
+  oricls_batch_size: 1
+  recovery: True
+  device: "gpu"
+
+######################################## Support ########################################

requirements.txt

@@ -23,3 +23,11 @@ lxml
 openpyxl
 premailer
 PyMuPDF
+######## For Chatocrv3 #######
+qianfan==0.0.3
+langchain==0.1.5
+langchain-community==0.0.17
+erniebot == 0.5.0
+erniebot-agent == 0.5.0
+unstructured
+networkx