Update Dockerfile

Dockerfile

@@ -3,7 +3,12 @@ WORKDIR /code
 COPY . .
 
 # Install C++ build tools first
-RUN apt-get update && apt-get install -y build-essential && apt-get clean
+# RUN apt-get update && apt-get install -y build-essential && apt-get clean
+
+RUN apt-get update && \
+    apt-get install -y build-essential libgl1-mesa-glx libglib2.0-0 && \
+    apt-get clean
+
 
 # Install all dependencies (including pillow)
 RUN pip install --upgrade pip && pip install -r requirements.txt
@@ -14,7 +19,14 @@ ENV TRANSFORMERS_CACHE="/data"
 RUN mkdir -p /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')"
+# RUN python -c "from transformers import BlipProcessor, BlipForConditionalGeneration; BlipProcessor.from_pretrained('Salesforce/blip-image-captioning-base'); BlipForConditionalGeneration.from_pretrained('Salesforce/blip-image-captioning-base')"
+
+# Pre-download DeepFace models
+RUN python -c "from deepface import DeepFace; DeepFace.analyze('https://raw.githubusercontent.com/serengil/deepface/master/tests/dataset/img1.jpg', actions=['age', 'gender', 'race', 'emotion'], enforce_detection=False)"
+# Pre-download InsightFace models
+RUN python -c \"import insightface; import numpy as np; app = insightface.app.FaceAnalysis(name='buffalo_l', providers=['CPUExecutionProvider']); app.prepare(ctx_id=0); img = np.zeros((640, 640, 3), dtype=np.uint8); app.get(img)\"
+
+
 
 EXPOSE 7860
 CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]