Update Dockerfile

Dockerfile

@@ -22,10 +22,22 @@ RUN pip install --upgrade pip && pip install -r requirements.txt
 # ENV DEEPFACE_HOME="/data/.deepface"
 # RUN mkdir -p /data/.deepface
 
+# ENV HF_HOME="/data"
+# ENV TRANSFORMERS_CACHE="/data"
+# ENV DEEPFACE_HOME="/data"
+# RUN mkdir -p /data && chmod -R 777 /data
+# ENV INSIGHTFACE_HOME="/data/.insightface"
+# RUN mkdir -p /data/.insightface && chmod -R 777 /data/.insightface
+
+
 ENV HF_HOME="/data"
 ENV TRANSFORMERS_CACHE="/data"
 ENV DEEPFACE_HOME="/data"
-RUN mkdir -p /data && chmod -R 777 /data
+ENV INSIGHTFACE_HOME="/data/.insightface"
+RUN mkdir -p /data && \
+    mkdir -p /data/.insightface && \
+    chmod -R 777 /data
+
 # (Optional, but speeds up runtime) Pre-download BLIP model weights
 # RUN python -c "from transformers import BlipProcessor, BlipForConditionalGeneration; BlipProcessor.from_pretrained('Salesforce/blip-image-captioning-base'); BlipForConditionalGeneration.from_pretrained('Salesforce/blip-image-captioning-base')"