diff --git a/requirements.txt b/requirements.txt
index 0636b3d..f9151af 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,4 +1,4 @@
-torch==2.4.1
+torch==2.4.1 ## for fastai
 torchaudio==2.4.1
 torchvision==0.19.1
 python-magic==0.4.27
@@ -14,5 +14,11 @@ alembic==1.13.3
 black==24.10.0
 Flask-SQLAlchemy==3.1.1
 bcrypt==4.2.0
-tensorflow==2.17.0
+tensorflow[and-cuda]==2.17.0
 transformers==4.45.2
+scikit-learn==1.5.2
+fastai==2.7.17
+# bark==0.1.5
+joblib==1.4.2
+dspy-ai==2.5.12
+pgvector==0.3.5