Not-Grim-Refer commited on
Commit
0bd3785
·
1 Parent(s): 3ced0ae

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +6 -1
requirements.txt CHANGED
@@ -1,3 +1,8 @@
1
  gradio
2
  transformers==4.29.0
3
- torch==1.7.1+cpu
 
 
 
 
 
 
1
  gradio
2
  transformers==4.29.0
3
+ torch>=2.0
4
+ datasets==2.9.0
5
+ accelerate==0.17.1
6
+ evaluate==0.4.0
7
+ tensorboard
8
+ scikit-learn