Gbssreejith commited on
Commit
bbd227d
1 Parent(s): 4172b6a

Update requirements.py

Browse files
Files changed (1) hide show
  1. requirements.txt +8 -3
requirements.txt CHANGED
@@ -1,5 +1,10 @@
 
 
 
 
 
 
 
 
1
  numpy
2
- torch
3
  nltk
4
- scikit-learn
5
- pandas
 
1
+ torch==2.0.0+cu118
2
+ torchvision==0.15.0+cu118
3
+ torchaudio==2.0.0+cu118
4
+ -f https://download.pytorch.org/whl/torch_stable.html
5
+ transformers
6
+ gradio
7
+ pandas
8
+ scikit-learn
9
  numpy
 
10
  nltk