feat: faster docker build

cyse7125-su24-team10 · Aug 12, 2024 · 8975a1b · 8975a1b
1 parent 8c6123b
commit 8975a1b
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/Dockerfile b/Dockerfile
@@ -31,6 +31,10 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
 
 COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
+RUN pip install --no-cache-dir --upgrade pip
+RUN pip install --no-cache-dir torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
+RUN pip install --no-cache-dir sentence-transformers
+
 
 # Final stage
 FROM python:3.9-slim

diff --git a/requirements.txt b/requirements.txt
@@ -28,4 +28,4 @@ langchain
 langchain_community
 faiss-cpu
 langchain-groq
-sentence_transformers
+#sentence_transformers