ochsncon commited on
Commit
768b248
·
verified ·
1 Parent(s): bcbe62f

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +4 -2
requirements.txt CHANGED
@@ -1,4 +1,4 @@
1
- gradio
2
  pandas
3
  numpy
4
  scikit-learn
@@ -10,4 +10,6 @@ python-dotenv
10
  datasets
11
  evaluate
12
  transformers
13
- accelerate
 
 
 
1
+ gradio>=4.0
2
  pandas
3
  numpy
4
  scikit-learn
 
10
  datasets
11
  evaluate
12
  transformers
13
+ torch==2.2.2
14
+ torchvision==0.17.2
15
+ torchaudio==2.2.2