Thomas Chaigneau commited on
Commit
d0e2596
1 Parent(s): 1b5ff91

add torch to requirements

Browse files
Files changed (1) hide show
  1. requirements.txt +4 -1
requirements.txt CHANGED
@@ -1,2 +1,5 @@
1
  streamlit==1.3.1
2
- transformers==4.15.0
 
 
 
 
1
  streamlit==1.3.1
2
+ transformers==4.15.0
3
+ torch==1.10.1+cpu
4
+ torchaudio==0.10.1+cpu
5
+ torchvision==0.11.2+cpu