mashe commited on
Commit
537868a
·
verified ·
1 Parent(s): 16067a8

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +3 -3
Dockerfile CHANGED
@@ -23,11 +23,11 @@ EXPOSE 7860
23
  # Define environment variable for Gradio server port
24
  ENV GRADIO_SERVER_PORT=7860
25
 
26
- # Set the TRANSFORMERS_CACHE environment variable
27
- ENV TRANSFORMERS_CACHE=/app/.cache/huggingface/hub
28
 
29
  # Create the cache directory if it doesn't exist
30
- RUN mkdir -p $TRANSFORMERS_CACHE
31
 
32
  # Run app.py when the container launches
33
  CMD ["python", "app.py"]
 
23
  # Define environment variable for Gradio server port
24
  ENV GRADIO_SERVER_PORT=7860
25
 
26
+ # Set the HF_HOME environment variable to a writable directory
27
+ ENV HF_HOME=/app/.cache/huggingface
28
 
29
  # Create the cache directory if it doesn't exist
30
+ RUN mkdir -p $HF_HOME
31
 
32
  # Run app.py when the container launches
33
  CMD ["python", "app.py"]