Matthew Hollings commited on
Commit
8b85762
1 Parent(s): 3e7d8ba

add pytorch to requirements

Browse files
Files changed (1) hide show
  1. requirements.txt +4 -1
requirements.txt CHANGED
@@ -1,2 +1,5 @@
1
  transformers==4.11.3
2
- gradio==3.3.1
 
 
 
 
1
  transformers==4.11.3
2
+ gradio==3.3.1
3
+ torch==1.12.1
4
+ torchaudio==0.12.1
5
+ torchvision==0.13.1