Update Dockerfile

Dockerfile

@@ -1,35 +1,54 @@
-# ============================================================
-# GPU-ENABLED SELF-CONTAINED IMAGE FOR DEEPFAKE VIDEO DETECTOR
-# ============================================================
+# =======================================================
+# GPU-enabled base image (CUDA 12.1 + cuDNN 8)
+# =======================================================
 FROM pytorch/pytorch:2.2.2-cuda12.1-cudnn8-runtime
+
 ENV DEBIAN_FRONTEND=noninteractive \
     PYTHONDONTWRITEBYTECODE=1 \
     PYTHONUNBUFFERED=1
 
-# Force reinstall Gradio 4.44.1 every time (avoids cache)
+# =======================================================
+# Install Gradio first (pinned version)
+# =======================================================
 ARG GRADIO_FORCE_REINSTALL=1
 RUN pip install --no-cache-dir --upgrade gradio==4.44.1
 
-
-# System deps for OpenCV + dlib
+# =======================================================
+# System dependencies for OpenCV + dlib runtime
+# =======================================================
 RUN apt-get update && apt-get install -y --no-install-recommends \
-    build-essential cmake \
     libgl1 libglib2.0-0 libsm6 libxext6 libxrender1 \
     && rm -rf /var/lib/apt/lists/*
 
 WORKDIR /app
 
-# Copy app + requirements
+# =======================================================
+# Copy application files
+# =======================================================
 COPY requirements.txt /app/requirements.txt
 COPY app.py /app/app.py
-
-# ✅ Copy your model + landmark file directly into the image
 COPY models /app/models
 COPY shape_predictor_68_face_landmarks.dat /app/
 
-# Install TensorFlow (GPU-capable) + requirements
+# =======================================================
+# Install Python dependencies
+# =======================================================
 RUN pip install --no-cache-dir tensorflow==2.18.0
 RUN pip install --no-cache-dir -r requirements.txt
 
+# 🔧 Replace compiled dlib with prebuilt dlib-bin
+RUN pip uninstall -y dlib || true \
+ && pip install --no-cache-dir dlib-bin==19.24.2
+
+# =======================================================
+# Environment optimizations
+# =======================================================
+ENV OMP_NUM_THREADS=2 \
+    MKL_NUM_THREADS=2 \
+    OPENBLAS_NUM_THREADS=2 \
+    NUMEXPR_NUM_THREADS=2 \
+    TF_CPP_MIN_LOG_LEVEL=2 \
+    CUDA_VISIBLE_DEVICES=0
+
 EXPOSE 7860
 CMD ["python", "app.py"]