salohnana2018 commited on
Commit
c3bf42a
1 Parent(s): b96906d

Update requirements.txt

Browse files
Files changed (1) hide show
  1. 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+cu121 # or a compatible version if this still causes issues
5
  joblib==1.4.2
6
- scikit-learn==1.3.0 # or the latest stable version compatible with your setup
 
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