lgblkb commited on
Commit
3f6b923
·
1 Parent(s): 9e56aa7

fix: Add pytorch

Browse files
Files changed (1) hide show
  1. requirements.txt +4 -1
requirements.txt CHANGED
@@ -1,2 +1,5 @@
1
  streamlit~=1.37.0
2
- huggingface-hub~=0.24.5
 
 
 
 
1
  streamlit~=1.37.0
2
+ huggingface-hub~=0.24.5
3
+ torch==2.4.0+cpu
4
+ torchvision==0.19.0+cpu
5
+ torchaudio==2.4.0+cpu