KarthikAI commited on
Commit
51ee4c8
·
verified ·
1 Parent(s): a699516

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +10 -6
Dockerfile CHANGED
@@ -14,14 +14,18 @@ RUN apt-get update && \
14
  RUN pip install --upgrade pip && pip install -r requirements.txt
15
 
16
  # Set cache dir env vars (fixes Hugging Face Space permissions)
17
- ENV HF_HOME="/data"
18
- ENV TRANSFORMERS_CACHE="/data"
19
- RUN mkdir -p /data
20
 
21
- # Set DeepFace home directory (to avoid writing to /)
22
- ENV DEEPFACE_HOME="/data/.deepface"
23
- RUN mkdir -p /data/.deepface
24
 
 
 
 
 
25
  # (Optional, but speeds up runtime) Pre-download BLIP model weights
26
  # RUN python -c "from transformers import BlipProcessor, BlipForConditionalGeneration; BlipProcessor.from_pretrained('Salesforce/blip-image-captioning-base'); BlipForConditionalGeneration.from_pretrained('Salesforce/blip-image-captioning-base')"
27
 
 
14
  RUN pip install --upgrade pip && pip install -r requirements.txt
15
 
16
  # Set cache dir env vars (fixes Hugging Face Space permissions)
17
+ # ENV HF_HOME="/data"
18
+ # ENV TRANSFORMERS_CACHE="/data"
19
+ # RUN mkdir -p /data
20
 
21
+ # # Set DeepFace home directory (to avoid writing to /)
22
+ # ENV DEEPFACE_HOME="/data/.deepface"
23
+ # RUN mkdir -p /data/.deepface
24
 
25
+ ENV HF_HOME="/data"
26
+ ENV TRANSFORMERS_CACHE="/data"
27
+ ENV DEEPFACE_HOME="/data"
28
+ RUN mkdir -p /data && chmod -R 777 /data
29
  # (Optional, but speeds up runtime) Pre-download BLIP model weights
30
  # RUN python -c "from transformers import BlipProcessor, BlipForConditionalGeneration; BlipProcessor.from_pretrained('Salesforce/blip-image-captioning-base'); BlipForConditionalGeneration.from_pretrained('Salesforce/blip-image-captioning-base')"
31