banao-tech commited on
Commit
1a6ef75
·
verified ·
1 Parent(s): 32403bf

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +8 -10
requirements.txt CHANGED
@@ -1,10 +1,5 @@
1
  fastapi[all]
2
  uvicorn
3
- # PyTorch dependencies
4
- --extra-index-url https://download.pytorch.org/whl/cu121
5
- torch==2.1.2+cu121
6
- torchvision==0.16.2+cu121
7
-
8
  numpy
9
  opencv-python-headless
10
  transformers
@@ -12,8 +7,11 @@ python-multipart
12
  matplotlib
13
  easyocr
14
  paddleocr
15
- pip
16
- setuptools
17
- wheel
18
- paddlepaddle==2.5.1
19
- libssl1.1
 
 
 
 
1
  fastapi[all]
2
  uvicorn
 
 
 
 
 
3
  numpy
4
  opencv-python-headless
5
  transformers
 
7
  matplotlib
8
  easyocr
9
  paddleocr
10
+
11
+ # PyTorch CUDA 12.1 dependencies
12
+ --extra-index-url https://download.pytorch.org/whl/cu121
13
+ torch==2.1.2+cu121
14
+ torchvision==0.16.2+cu121
15
+
16
+ # PaddlePaddle dependencies
17
+ paddlepaddle==2.6.0