From 0345a80033accdc5ac211b5af99bd70ad84b5c37 Mon Sep 17 00:00:00 2001
From: "Timothy J. Baek" <timothyjrbeck@gmail.com>
Date: Sun, 7 Jan 2024 20:21:32 -0800
Subject: [PATCH] feat: include embedding weight to image

---
 Dockerfile | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/Dockerfile b/Dockerfile
index 88cfc6b93..3019c2eaa 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -28,6 +28,14 @@ WORKDIR /app/backend
 
 COPY ./backend/requirements.txt ./requirements.txt
 RUN pip3 install -r requirements.txt
+
+RUN MODEL_DIR="/root/.cache/chroma/onnx_models/all-MiniLM-L6-v2" &&\
+    ARCHIVE_NAME="onnx.tar.gz" &&\
+    mkdir -p $MODEL_DIR &&\
+    cd $MODEL_DIR &&\
+    wget -O $ARCHIVE_NAME "https://chroma-onnx-models.s3.amazonaws.com/all-MiniLM-L6-v2/$ARCHIVE_NAME" &&\
+    tar -xzf $ARCHIVE_NAME
+
 # RUN python -c "from sentence_transformers import SentenceTransformer; model = SentenceTransformer('all-MiniLM-L6-v2')"
 
 COPY ./backend .