Spaces:
Sleeping
Sleeping
update requirement
Browse files- requirements.txt +20 -3
requirements.txt
CHANGED
|
@@ -4,7 +4,6 @@
|
|
| 4 |
torch==2.3.1
|
| 5 |
torchvision==0.18.1
|
| 6 |
torchaudio==2.3.1
|
| 7 |
-
pytorch-cuda==12.1 # nếu dùng conda, pip sẽ bỏ qua
|
| 8 |
|
| 9 |
# ===============================
|
| 10 |
# Transformers Ecosystem
|
|
@@ -25,9 +24,27 @@ rouge-score>=0.1.2
|
|
| 25 |
# ===============================
|
| 26 |
# Semantic Search / FAISS
|
| 27 |
# ===============================
|
| 28 |
-
faiss-gpu
|
|
|
|
|
|
|
| 29 |
|
| 30 |
# ===============================
|
| 31 |
# PDF / Text Processing
|
| 32 |
# ===============================
|
| 33 |
-
PyMuPDF
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 4 |
torch==2.3.1
|
| 5 |
torchvision==0.18.1
|
| 6 |
torchaudio==2.3.1
|
|
|
|
| 7 |
|
| 8 |
# ===============================
|
| 9 |
# Transformers Ecosystem
|
|
|
|
| 24 |
# ===============================
|
| 25 |
# Semantic Search / FAISS
|
| 26 |
# ===============================
|
| 27 |
+
# Nếu GPU runtime (T4 / A10G) => dùng faiss-gpu
|
| 28 |
+
# Nếu CPU-only => đổi sang faiss-cpu
|
| 29 |
+
faiss-cpu==1.8.0
|
| 30 |
|
| 31 |
# ===============================
|
| 32 |
# PDF / Text Processing
|
| 33 |
# ===============================
|
| 34 |
+
PyMuPDF>=1.23.0
|
| 35 |
+
underthesea>=6.8.0
|
| 36 |
+
numpy>=1.26.0
|
| 37 |
+
pandas>=2.2.0
|
| 38 |
+
|
| 39 |
+
# ===============================
|
| 40 |
+
# Web / API
|
| 41 |
+
# ===============================
|
| 42 |
+
fastapi>=0.111.0
|
| 43 |
+
uvicorn[standard]>=0.30.0
|
| 44 |
+
|
| 45 |
+
# ===============================
|
| 46 |
+
# Utils
|
| 47 |
+
# ===============================
|
| 48 |
+
regex>=2024.4.16
|
| 49 |
+
tqdm>=4.66.3
|
| 50 |
+
requests>=2.31.0
|