thechaiexperiment commited on
Commit
ffb1d75
·
1 Parent(s): b16ca60

Update Dockerfile

Browse files
Files changed (1) hide show
  1. 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 # Adding HF_HOME for 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')"
 
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')"