fix: dockerfile

02f364bf · Timothy J. Baek · 706a8d4f · 02f364bf · 02f364bf
Commit 02f364bf authored Jan 07, 2024 by Timothy J. Baek
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 7 deletions

Dockerfile Dockerfile +2 -0

backend/requirements.txt backend/requirements.txt +0 -7

No files found.
--- 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')"

--- 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