Update requirements
Browse files- requirements +3 -1
requirements
CHANGED
|
@@ -3,4 +3,6 @@ transformers>=4.40.0
|
|
| 3 |
datasets>=2.18.0
|
| 4 |
sentence-transformers>=2.6.0
|
| 5 |
gradio>=4.24.0
|
| 6 |
-
scikit-learn>=1.4.0
|
|
|
|
|
|
|
|
|
| 3 |
datasets>=2.18.0
|
| 4 |
sentence-transformers>=2.6.0
|
| 5 |
gradio>=4.24.0
|
| 6 |
+
scikit-learn>=1.4.0
|
| 7 |
+
-torch
|
| 8 |
+
+torch==2.2.1+cpu --extra-index-url https://download.pytorch.org/whl/cpu
|