AkshayaKeerthi commited on
Commit
5150512
1 Parent(s): dbce713

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -0
Dockerfile CHANGED
@@ -15,6 +15,7 @@ RUN useradd -m -u 1000 user
15
 
16
  # Set the cache directory and pre-download the model
17
  RUN mkdir -p /app/.cache/huggingface
 
18
  ENV HF_HOME=/app/.cache/huggingface
19
 
20
  # Switch to the "user" user
 
15
 
16
  # Set the cache directory and pre-download the model
17
  RUN mkdir -p /app/.cache/huggingface
18
+ ENV TRANSFORMERS_CACHE=/app/.cache/huggingface
19
  ENV HF_HOME=/app/.cache/huggingface
20
 
21
  # Switch to the "user" user