Hugging Face's logo

Spaces:
KarthikAI
/
StickerGen
Sleeping

App Files Community
KarthikAI commited on
Commit
a699516
·
verified ·
1 Parent(s): 6fb7a18

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +4 -0
Dockerfile CHANGED
@@ -18,6 +18,10 @@ ENV HF_HOME="/data"
18
  ENV TRANSFORMERS_CACHE="/data"
19
  RUN mkdir -p /data
20
 
 
 
 
 
21
  # (Optional, but speeds up runtime) Pre-download BLIP model weights
22
  # RUN python -c "from transformers import BlipProcessor, BlipForConditionalGeneration; BlipProcessor.from_pretrained('Salesforce/blip-image-captioning-base'); BlipForConditionalGeneration.from_pretrained('Salesforce/blip-image-captioning-base')"
23
 
 
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