Spaces:
Sleeping
Sleeping
Commit
·
4deecd8
1
Parent(s):
ffb1d75
Update Dockerfile
Browse files- Dockerfile +3 -1
Dockerfile
CHANGED
@@ -26,7 +26,9 @@ RUN python -c "import nltk; nltk.download('punkt')"
|
|
26 |
# Create and set permissions for the Transformers cache directory
|
27 |
RUN mkdir -p /code/transformers_cache && chmod -R 777 /code/transformers_cache
|
28 |
ENV TRANSFORMERS_CACHE=/code/transformers_cache
|
29 |
-
|
|
|
|
|
30 |
|
31 |
# Download sentence-transformers model to avoid recreating it at runtime
|
32 |
RUN python -c "from sentence_transformers import SentenceTransformer; SentenceTransformer('sentence-transformers/all-MiniLM-L6-v2')"
|
|
|
26 |
# Create and set permissions for the Transformers cache directory
|
27 |
RUN mkdir -p /code/transformers_cache && chmod -R 777 /code/transformers_cache
|
28 |
ENV TRANSFORMERS_CACHE=/code/transformers_cache
|
29 |
+
|
30 |
+
# Set HF_HOME for Hugging Face cache
|
31 |
+
ENV HF_HOME=/code/transformers_cache
|
32 |
|
33 |
# Download sentence-transformers model to avoid recreating it at runtime
|
34 |
RUN python -c "from sentence_transformers import SentenceTransformer; SentenceTransformer('sentence-transformers/all-MiniLM-L6-v2')"
|