diff --git a/Dockerfile b/Dockerfile
index 8e81b7749..64a307534 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -40,6 +40,8 @@ COPY --from=build /app/build /app/build
 WORKDIR /app/backend
 
 COPY ./backend/requirements.txt ./requirements.txt
+
+RUN pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
 RUN pip3 install -r requirements.txt
 
 # RUN python -c "from sentence_transformers import SentenceTransformer; model = SentenceTransformer('all-MiniLM-L6-v2')"
diff --git a/backend/requirements.txt b/backend/requirements.txt
index e0c2f25a3..06af0efc6 100644
--- a/backend/requirements.txt
+++ b/backend/requirements.txt
@@ -16,13 +16,6 @@ aiohttp
 peewee
 bcrypt
 
---index-url https://download.pytorch.org/whl/cpu
-torch 
---index-url https://download.pytorch.org/whl/cpu
-torchvision 
---index-url https://download.pytorch.org/whl/cpu
-torchaudio 
-
 langchain
 langchain-community
 chromadb