Browse Source

update requirements and setup

赵小蒙 1 year ago
parent
commit
3aa8ccdceb
2 changed files with 2 additions and 3 deletions
  1. 0 1
      requirements.txt
  2. 2 2
      setup.py

+ 0 - 1
requirements.txt

@@ -14,5 +14,4 @@ wordninja>=2.0.0
 scikit-learn>=1.0.2
 nltk==3.8.1
 s3pathlib>=2.1.1
-paddleocr @ https://github.com/magicpdf/PaddleOCR/releases/download/paddleocr-2.8.2-released/paddleocr-2.8.2-py3-none-any.whl
 pdfminer.six>=20231228

+ 2 - 2
setup.py

@@ -29,8 +29,8 @@ if __name__ == '__main__':
         packages=find_packages(),  # 包含所有的包
         install_requires=parse_requirements('requirements.txt'),  # 项目依赖的第三方库
         extras_require={
-            "gpu": ["paddlepaddle-gpu"],
-            "cpu": ["paddlepaddle"],
+            "gpu": ["paddleocr", "paddlepaddle-gpu"],
+            "cpu": ["paddleocr", "paddlepaddle"],
         },
         description="A practical tool for converting PDF to Markdown",  # 简短描述
         long_description=long_description,  # 详细描述