소스 검색

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")