idkash1 commited on
Commit
a042ddf
·
verified ·
1 Parent(s): ebea54d

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +3 -2
Dockerfile CHANGED
@@ -12,8 +12,9 @@ RUN mkdir -p /cache/matplotlib \
12
  && chmod -R 777 /cache/matplotlib
13
 
14
  # Set environment variables for Hugging Face and Matplotlib
15
- ENV HF_HOME=/cache/huggingface
16
- ENV MPLCONFIGDIR=/cache/matplotlib
 
17
 
18
  # Flask
19
  ENV FLASK_APP=app.py
 
12
  && chmod -R 777 /cache/matplotlib
13
 
14
  # Set environment variables for Hugging Face and Matplotlib
15
+ ENV HF_HOME=/tmp/cacheHuggingface/PHI2
16
+ ENV TRANSFORMERS_CACHE=/tmp/cacheHuggingface/PHI2
17
+ ENV MPLCONFIGDIR=/tmp/cache/matplotlib
18
 
19
  # Flask
20
  ENV FLASK_APP=app.py