Merge pull request #155 from i-dot-ai/bugfix/streamlit-app-model

added EMBEDDING_MODEL to streamlit-app build
i-dot-ai · Mar 27, 2024 · 9560d5a · 9560d5a
2 parents 12ba099 + 2ef8020
commit 9560d5a
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/docker-compose.yml b/docker-compose.yml
@@ -43,6 +43,8 @@ services:
     build:
       context: .
       dockerfile: ./streamlit_app/Dockerfile
+      args:
+        - EMBEDDING_MODEL=${EMBEDDING_MODEL:-}
     env_file:
       - .env
     ports:

diff --git a/streamlit_app/Dockerfile b/streamlit_app/Dockerfile
@@ -14,6 +14,7 @@ ADD pyproject.toml poetry.lock ./
 RUN --mount=type=cache,target=$POETRY_CACHE_DIR  poetry install --no-root --no-ansi --with streamlit-app,ai,ingester --without dev,worker,api  --no-root
 
 FROM python:3.11-slim-buster as runtime
+ARG EMBEDDING_MODEL
 
 RUN apt-get update
 RUN apt-get install -y libgl-dev libmagic-dev inetutils-ping
@@ -27,8 +28,7 @@ COPY --from=builder ${VIRTUAL_ENV} ${VIRTUAL_ENV}
 
 ADD redbox/ /app/redbox
 ADD download_embedder.py /app/
-RUN python download_embedder.py
-
+RUN python download_embedder.py --model_name ${EMBEDDING_MODEL}
 ADD streamlit_app/ /app
 
 EXPOSE 8501