From 3aa8ccdceb7c42e1463ecd78d01a1b1946dab2b1 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E8=B5=B5=E5=B0=8F=E8=92=99?= <moe@myhloli.com>
Date: Tue, 25 Jun 2024 19:05:42 +0800
Subject: [PATCH] update requirements and setup

---
 requirements.txt | 1 -
 setup.py         | 4 ++--
 2 files changed, 2 insertions(+), 3 deletions(-)

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,  # 详细描述