Samizie commited on
Commit
9c98458
·
verified ·
1 Parent(s): f3a1902

Update Dockerfile

Browse files
Files changed (1) hide show
  1. 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['TRANSFORMERS_CACHE'] = '/app/.cache'; os.environ['HF_HOME'] = '/app/.cache'; from sentence_transformers import SentenceTransformer; SentenceTransformer('sentence-transformers/all-MiniLM-L6-v2', cache_folder='/app/.cache')"
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