diff --git a/requirements.txt b/requirements.txt index d05b93af..a6dc5241 100644 --- a/requirements.txt +++ b/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 \ No newline at end of file diff --git a/setup.py b/setup.py index 75b00052..c71aee68 100644 --- a/setup.py +++ b/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, # 详细描述