diff --git a/requirements.txt b/requirements.txt
index aa91a4e..a860dfc 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,3 +1,8 @@
 transformers>=4.19.2
 einops>=0.4.1
-timm>=0.4.12
\ No newline at end of file
+timm>=0.4.12
+torch
+towhee.models
+torch
+torchvision
+towhee
\ No newline at end of file