瀏覽代碼

fix chatocr name

gaotingquan 1 年之前
父節點
當前提交
8a8067f238
共有 2 個文件被更改,包括 2 次插入1 次删除
  1. 1 1
      paddlex/inference/pipelines/ppchatocrv3/ppchatocrv3.py
  2. 1 0
      setup.py

+ 1 - 1
paddlex/inference/pipelines/ppchatocrv3/ppchatocrv3.py

@@ -40,7 +40,7 @@ PROMPT_FILE = os.path.join(os.path.dirname(__file__), "ch_prompt.yaml")
 class PPChatOCRPipeline(TableRecPipeline):
     """PP-ChatOCRv3 Pileline"""
 
-    entities = "chatocrv3"
+    entities = "PP-ChatOCRv3"
 
     def __init__(
         self,

+ 1 - 0
setup.py

@@ -65,6 +65,7 @@ def packages_and_package_data():
         Path(p).relative_to("paddlex").as_posix()
         for p in glob.glob("paddlex/pipelines/*.yaml")
     ]
+    pkg_data.append("inference/pipelines/ppchatocrv3/ch_prompt.yaml")
     pkg_data.extend(pipeline_config)
     pkg_data.append(".version")
     pkg_data.append("utils/fonts/PingFang-SC-Regular.ttf")