rename_docprompt_pipeline (PaddlePaddle#3731)

huchuqi12 · Nov 14, 2022 · b63f356 · b63f356
1 parent 70c8bfc
commit b63f356
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 7 deletions.
diff --git a/pipelines/examples/document-intelligence/docprompt_example.py b/pipelines/examples/document-intelligence/docprompt_example.py
@@ -34,7 +34,7 @@ def docprompt_pipeline():
 
     preprocessor = DocOCRProcessor(use_gpu=use_gpu)
     docprompter = DocPrompter(use_gpu=use_gpu, batch_size=args.batch_size)
-    pipe = DocPipeline(preprocessor=preprocessor, modelrunner=docprompter)
+    pipe = DocPipeline(preprocessor=preprocessor, docreader=docprompter)
     # image link input
     meta = {
         "doc":

diff --git a/pipelines/pipelines/pipelines/standard_pipelines.py b/pipelines/pipelines/pipelines/standard_pipelines.py
@@ -272,17 +272,17 @@ class DocPipeline(BaseStandardPipeline):
     Pipeline for document intelligence.
     """
 
-    def __init__(self, preprocessor: BaseComponent, modelrunner: BaseComponent):
+    def __init__(self, preprocessor: BaseComponent, docreader: BaseComponent):
         """
         :param preprocessor: file/image preprocessor instance
-        :param modelrunner: document model runner instance
+        :param docreader: document model runner instance
         """
         self.pipeline = Pipeline()
         self.pipeline.add_node(component=preprocessor,
                                name="PreProcessor",
                                inputs=["Query"])
-        self.pipeline.add_node(component=modelrunner,
-                               name="Runner",
+        self.pipeline.add_node(component=docreader,
+                               name="Reader",
                                inputs=["PreProcessor"])
 
     def run(self,

diff --git a/pipelines/rest_api/pipeline/docprompt.yaml b/pipelines/rest_api/pipeline/docprompt.yaml
@@ -6,7 +6,7 @@ components:
           use_gpu: True
           lang: ch
     type: DocOCRProcessor
-  - name: Runner
+  - name: Reader
     params:
           topn: 1
           use_gpu: True
@@ -23,7 +23,7 @@ pipelines:
     nodes:
       - name: PreProcessor
         inputs: [Query]
-      - name: Runner
+      - name: Reader
         inputs: [PreProcessor]