Update Dockerfile
Browse files- Dockerfile +1 -1
Dockerfile
CHANGED
@@ -51,7 +51,7 @@ RUN mkdir -p /app/.cache && \
|
|
51 |
RUN chmod -R 755 /app
|
52 |
|
53 |
# Preload SentenceTransformer model
|
54 |
-
RUN python -c "import os; os.environ['
|
55 |
|
56 |
# Expose port
|
57 |
EXPOSE 7860
|
|
|
51 |
RUN chmod -R 755 /app
|
52 |
|
53 |
# Preload SentenceTransformer model
|
54 |
+
RUN python -c "import os; os.environ['HF_HOME'] = '/workspace/.cache'; from sentence_transformers import SentenceTransformer; SentenceTransformer('sentence-transformers/all-MiniLM-L6-v2', cache_folder='/workspace/.cache')"
|
55 |
|
56 |
# Expose port
|
57 |
EXPOSE 7860
|