shounakpaul95 commited on
Commit
54df980
1 Parent(s): 9c31c29

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +3 -1
requirements.txt CHANGED
@@ -1,5 +1,6 @@
1
  seaborn
2
  scipy
 
3
  datasets==2.14.5
4
  gradio
5
  huggingface-hub==0.18.0
@@ -14,4 +15,5 @@ transformers==4.17.0
14
  scikit-learn==1.0.2
15
  py-rouge==1.1
16
  bert-score==0.3.13
17
- sentencepiece==0.1.99
 
 
1
  seaborn
2
  scipy
3
+ pytorch-cuda
4
  datasets==2.14.5
5
  gradio
6
  huggingface-hub==0.18.0
 
15
  scikit-learn==1.0.2
16
  py-rouge==1.1
17
  bert-score==0.3.13
18
+ sentencepiece==0.1.99
19
+ tokenizers