add torch to requirments

BSalita · Oct 12, 2023 · 712fec4 · 712fec4
1 parent c74f02d
commit 712fec4
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -11,11 +11,13 @@ termcolor
 streamlit-chat
 bs4
 endplay
-#torch
-scikit-learn
 sqlalchemy
 sqlalchemy_utils
 safetensors
+scikit-learn
+torch
+torchvision
+torchaudio
 
 # for streamlitlib
 reportlab