Vallabhpatil777 commited on
Commit
ee7fa56
·
verified ·
1 Parent(s): e391c27

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +6 -0
Dockerfile CHANGED
@@ -14,6 +14,12 @@ COPY src/ ./src/
14
  COPY main.py ./
15
  COPY . .
16
 
 
 
 
 
 
 
17
 
18
  RUN pip3 install -r requirements.txt
19
 
 
14
  COPY main.py ./
15
  COPY . .
16
 
17
+ ENV HF_HOME=/app/hf_cache
18
+ ENV TRANSFORMERS_CACHE=/app/hf_cache
19
+ ENV XDG_CACHE_HOME=/app/hf_cache
20
+
21
+ RUN mkdir -p /app/hf_cache
22
+
23
 
24
  RUN pip3 install -r requirements.txt
25