Spaces:
Sleeping
Sleeping
salohnana2018
commited on
Commit
•
c3bf42a
1
Parent(s):
b96906d
Update requirements.txt
Browse files- requirements.txt +2 -2
requirements.txt
CHANGED
@@ -1,6 +1,6 @@
|
|
1 |
gradio==4.42.0
|
2 |
pandas==2.1.4
|
3 |
numpy==1.26.4
|
4 |
-
torch==2.3.1
|
5 |
joblib==1.4.2
|
6 |
-
scikit-learn==1.3.0
|
|
|
1 |
gradio==4.42.0
|
2 |
pandas==2.1.4
|
3 |
numpy==1.26.4
|
4 |
+
torch==2.3.1 # Or torch==2.3.1+cu118 for CUDA 11.8 support
|
5 |
joblib==1.4.2
|
6 |
+
scikit-learn==1.3.0
|