add requirement

3 years ago · 87fc5edf83
1 changed files with 4 additions and 0 deletions
--- a/requirements.txt
+++ b/requirements.txt
@ -1,2 +1,6 @@
 numpy
 timm>=0.5.4
+
+torch
+towhee
+torchvision