Spaces:
Sleeping
Sleeping
Commit
·
ffb1d75
1
Parent(s):
b16ca60
Update Dockerfile
Browse files- Dockerfile +1 -1
Dockerfile
CHANGED
@@ -26,7 +26,7 @@ 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 |
-
ENV HF_HOME=/code/transformers_cache #
|
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 |
+
ENV HF_HOME=/code/transformers_cache # Hugging Face cache
|
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')"
|