1 parent fe321f3 commit 5da8fa2Copy full SHA for 5da8fa2
1 file changed
api/Dockerfile
@@ -7,7 +7,7 @@ RUN apt-get update && apt-get install -y --no-install-recommends gcc libpq-dev &
7
COPY requirements.txt .
8
RUN pip install --no-cache-dir --index-url https://download.pytorch.org/whl/cpu torch==2.3.1
9
RUN pip install --no-cache-dir -r requirements.txt
10
-RUN python -c "from sentence_transformers import SentenceTransformer; SentenceTransformer('all-MiniLM-L6-v2')"
+RUN python -c "from sentence_transformers import SentenceTransformer; SentenceTransformer('BAAI/bge-small-en-v1.5')"
11
12
COPY . /app/api
13
0 commit comments