.

mlibrary · May 2, 2024 · 0499d37 · 0499d37
1 parent a559385
commit 0499d37
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/py/Dockerfile b/py/Dockerfile
@@ -2,14 +2,17 @@ FROM python:3.12
 
 RUN apt-get update && \
   apt-get install -yqq --no-install-recommends \
-    cargo && \
+    cargo vim && \
   rm -rf \
     /var/lib/apt/lists/* \
     /usr/share/doc \
     /usr/share/locale \
     /usr/share/man
 
-RUN mkdir -p /app
+RUN mkdir -p /app/cache/pip /app/cache/hf
+
+RUN HF_HOME=/app/cache/hf PIP_CACHE_DIR=/app/cache/pip pip install -U sentence-transformers && \
+  rm -rf /app/cache
 
 WORKDIR /app