CsanadT commited on
Commit
556e022
1 Parent(s): 4f15dfc

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +3 -1
requirements.txt CHANGED
@@ -3,5 +3,7 @@ scikit-learn
3
  hopsworks
4
  joblib
5
  transformers
6
- tensorflow
 
 
7
  ffmpeg
 
3
  hopsworks
4
  joblib
5
  transformers
6
+ torch
7
+ torchvision
8
+ torchaudio
9
  ffmpeg