diff --git a/Dockerfile b/Dockerfile
index 85d210a..4ec78ab 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -12,4 +12,10 @@ ENV PYTHONDONTWRITEBYTECODE 1
 # the stdout and stderr streams are sent straight to terminal (e.g. your container log) without being first buffered and that you can see the output of your application (e.g. django logs) in real tim
 ENV PYTHONUNBUFFERED 1
 
+# Insalling ML dependencies inside requirements.txt
+COPY requirements.txt /tmp/pip-tmp/
+RUN pip --disable-pip-version-check --no-cache-dir install -r /tmp/pip-tmp/requirements.txt \
+   && rm -rf /tmp/pip-tmp
+
+
 RUN pip --disable-pip-version-check --no-cache-dir install -r https://raw.githubusercontent.com/SkywardAI/chat-backend/main/backend/requirements.txt
diff --git a/requirements.txt b/requirements.txt
new file mode 100644
index 0000000..7e43509
--- /dev/null
+++ b/requirements.txt
@@ -0,0 +1,3 @@
+torch == 2.2.2
+transformers == 4.39.2
+sentencepiece == 0.2.0