prd101-wd commited on
Commit
4ff3c05
·
verified ·
1 Parent(s): b310977

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +7 -0
Dockerfile CHANGED
@@ -1,5 +1,12 @@
1
  FROM python:3.9-slim
2
 
 
 
 
 
 
 
 
3
  WORKDIR /app
4
 
5
  RUN apt-get update && apt-get install -y \
 
1
  FROM python:3.9-slim
2
 
3
+ # Set environment variables to avoid caching issues with Hugging Face
4
+ ENV HF_HOME=/tmp/huggingface \
5
+ TRANSFORMERS_CACHE=/tmp/huggingface/transformers \
6
+ HF_DATASETS_CACHE=/tmp/huggingface/datasets \
7
+ HUGGINGFACE_HUB_CACHE=/tmp/huggingface/hub \
8
+ XDG_CACHE_HOME=/tmp/huggingface
9
+
10
  WORKDIR /app
11
 
12
  RUN apt-get update && apt-get install -y \