Somalitts commited on
Commit
ad92abc
·
verified ·
1 Parent(s): b9565e2

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +5 -0
Dockerfile CHANGED
@@ -7,4 +7,9 @@ RUN pip install --no-cache-dir -r requirements.txt
7
 
8
  COPY . .
9
 
 
 
 
 
 
10
  CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
 
7
 
8
  COPY . .
9
 
10
+ # Use safe cache dirs
11
+ ENV TRANSFORMERS_CACHE=/tmp/hf-cache
12
+ ENV HF_HOME=/tmp/hf-home
13
+ ENV XDG_CACHE_HOME=/tmp/xdg
14
+
15
  CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]