diff --git a/setup.py b/setup.py
index 33e7084..1d989d5 100644
--- a/setup.py
+++ b/setup.py
@@ -37,9 +37,13 @@
 )
 
 required = [
-    "click",
-    "tqdm",
     "colorlog",
+    "click",
+    "numpy",
+    "torch",
+    "torchvision",
+    "transformers",
+    "pillow",
     "chromadb",
 ]
 extras_require = {