Spaces:
Running
Running
Update requirements.txt
Browse files- requirements.txt +9 -6
requirements.txt
CHANGED
|
@@ -2,14 +2,20 @@
|
|
| 2 |
gradio==4.44.0
|
| 3 |
|
| 4 |
# Deep Learning Frameworks (CPU only)
|
|
|
|
| 5 |
torch==2.1.0+cpu
|
| 6 |
--extra-index-url https://download.pytorch.org/whl/cpu
|
| 7 |
|
| 8 |
-
# Transformers and NLP
|
| 9 |
transformers==4.37.0
|
| 10 |
tokenizers>=0.15.0
|
| 11 |
TorchCRF==1.1.0
|
| 12 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 13 |
# Computer Vision
|
| 14 |
ultralytics==8.0.196
|
| 15 |
opencv-python-headless==4.8.1.78
|
|
@@ -28,8 +34,5 @@ numpy==1.24.3
|
|
| 28 |
scipy==1.11.4
|
| 29 |
scikit-learn>=1.3.0
|
| 30 |
|
| 31 |
-
|
| 32 |
-
|
| 33 |
-
pix2text
|
| 34 |
-
optimum
|
| 35 |
-
onnxruntime
|
|
|
|
| 2 |
gradio==4.44.0
|
| 3 |
|
| 4 |
# Deep Learning Frameworks (CPU only)
|
| 5 |
+
# Ensure this block is installed with the correct index URL.
|
| 6 |
torch==2.1.0+cpu
|
| 7 |
--extra-index-url https://download.pytorch.org/whl/cpu
|
| 8 |
|
| 9 |
+
# Transformers and NLP
|
| 10 |
transformers==4.37.0
|
| 11 |
tokenizers>=0.15.0
|
| 12 |
TorchCRF==1.1.0
|
| 13 |
|
| 14 |
+
# Optimum and ONNX Runtime (Critical for Pix2Text)
|
| 15 |
+
# Pin these to known compatible versions to avoid the conflict.
|
| 16 |
+
optimum==1.17.1
|
| 17 |
+
onnxruntime==1.16.3
|
| 18 |
+
|
| 19 |
# Computer Vision
|
| 20 |
ultralytics==8.0.196
|
| 21 |
opencv-python-headless==4.8.1.78
|
|
|
|
| 34 |
scipy==1.11.4
|
| 35 |
scikit-learn>=1.3.0
|
| 36 |
|
| 37 |
+
# Pix2Text (Pin to a stable version)
|
| 38 |
+
pix2text==1.2.0
|
|
|
|
|
|
|
|
|