bichnhan2701 commited on
Commit
061d0f9
·
1 Parent(s): f84ed9c

Update dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -0
Dockerfile CHANGED
@@ -4,6 +4,8 @@ ENV PYTHONDONTWRITEBYTECODE=1 \
4
  PYTHONUNBUFFERED=1 \
5
  TMP_DIR=/tmp/uploads \
6
  PORT=7860
 
 
7
 
8
  # system deps (single RUN to minimize layers)
9
  RUN apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y \
 
4
  PYTHONUNBUFFERED=1 \
5
  TMP_DIR=/tmp/uploads \
6
  PORT=7860
7
+ ENV HF_HOME=/tmp/huggingface
8
+ ENV TRANSFORMERS_CACHE=/tmp/huggingface
9
 
10
  # system deps (single RUN to minimize layers)
11
  RUN apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y \