docker: pre-download transformer weights

This commit is contained in:
Timothy J. Baek 2024-01-07 03:39:02 -08:00
parent bf43415d12
commit 31bd6dfc15

View File

@ -28,6 +28,7 @@ WORKDIR /app/backend
COPY ./backend/requirements.txt ./requirements.txt COPY ./backend/requirements.txt ./requirements.txt
RUN pip3 install -r requirements.txt RUN pip3 install -r requirements.txt
RUN python -c "from sentence_transformers import SentenceTransformer; model = SentenceTransformer('all-MiniLM-L6-v2')"
COPY ./backend . COPY ./backend .