supertskone commited on
Commit
8c45fab
·
verified ·
1 Parent(s): 7018857

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +3 -2
Dockerfile CHANGED
@@ -17,10 +17,11 @@ RUN pip install --no-cache-dir --upgrade pip
17
  RUN pip install --no-cache-dir -r requirements.txt
18
 
19
  # Set environment variable for Hugging Face cache
20
- ENV HF_HOME=/app/cache
21
 
22
  # Create the cache directory
23
- RUN mkdir -p /app/cache
 
24
 
25
  # Expose port 8501 for Streamlit and port 5000 for Flask
26
  EXPOSE 8501
 
17
  RUN pip install --no-cache-dir -r requirements.txt
18
 
19
  # Set environment variable for Hugging Face cache
20
+ ENV TRANSFORMERS_CACHE=/app/cache
21
 
22
  # Create the cache directory
23
+ RUN mkdir -p /app/cache/hub
24
+ RUN chmod -R 777 /app/cache
25
 
26
  # Expose port 8501 for Streamlit and port 5000 for Flask
27
  EXPOSE 8501