philipobiorah commited on
Commit
dca576e
·
verified ·
1 Parent(s): 211e5c4

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -2
Dockerfile CHANGED
@@ -12,10 +12,10 @@ RUN apt-get update && apt-get install -y \
12
  && rm -rf /var/lib/apt/lists/*
13
 
14
  # Create writable cache directories within /tmp
15
- RUN mkdir -p /tmp/transformers_cache /tmp/matplotlib
16
 
17
  # Set environment variables for Hugging Face Transformers and Matplotlib
18
- ENV TRANSFORMERS_CACHE=/tmp/transformers_cache
19
  ENV MPLCONFIGDIR=/tmp/matplotlib
20
 
21
  # Copy requirements.txt first to leverage Docker caching
 
12
  && rm -rf /var/lib/apt/lists/*
13
 
14
  # Create writable cache directories within /tmp
15
+ RUN mkdir -p /tmp/huggingface_cache /tmp/matplotlib
16
 
17
  # Set environment variables for Hugging Face Transformers and Matplotlib
18
+ ENV HF_HOME=/tmp/huggingface_cache
19
  ENV MPLCONFIGDIR=/tmp/matplotlib
20
 
21
  # Copy requirements.txt first to leverage Docker caching