abdalraheemdmd commited on
Commit
151afba
·
verified ·
1 Parent(s): a950115

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +6 -1
Dockerfile CHANGED
@@ -2,10 +2,15 @@ FROM python:3.9
2
 
3
  WORKDIR /app
4
 
 
 
 
 
 
5
  # ✅ Install dependencies
6
  COPY requirements.txt .
7
  RUN pip install --no-cache-dir -r requirements.txt
8
- RUN pip install --no-cache-dir accelerate scipy
9
 
10
  COPY . .
11
 
 
2
 
3
  WORKDIR /app
4
 
5
+ # ✅ Set Hugging Face Cache to /tmp
6
+ ENV HF_HOME=/tmp/huggingface
7
+ ENV TRANSFORMERS_CACHE=/tmp/huggingface
8
+ ENV HF_HUB_CACHE=/tmp/huggingface
9
+
10
  # ✅ Install dependencies
11
  COPY requirements.txt .
12
  RUN pip install --no-cache-dir -r requirements.txt
13
+ RUN pip install --no-cache-dir accelerate scipy huggingface_hub
14
 
15
  COPY . .
16