Ver Fonte

fix chatocr name

gaotingquan há 1 ano atrás
pai
commit
8a8067f238
2 ficheiros alterados com 2 adições e 1 exclusões
  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")