Update Dockerfile

Dockerfile

@@ -1,58 +1,36 @@
-# ------------------------------------------------------
-# ✅ BASE IMAGE: Includes PyTorch + CUDA + cuDNN runtime
-# ------------------------------------------------------
+# ============================================================
+# GPU-ENABLED SELF-CONTAINED IMAGE FOR DEEPFAKE VIDEO DETECTOR
+# ============================================================
 FROM pytorch/pytorch:2.2.2-cuda12.1-cudnn8-runtime
 
-# ------------------------------------------------------
-# ✅ Set working directory
-# ------------------------------------------------------
-WORKDIR /app
-COPY . .
+ENV DEBIAN_FRONTEND=noninteractive \
+    PYTHONDONTWRITEBYTECODE=1 \
+    PYTHONUNBUFFERED=1
+
+# Force reinstall Gradio 4.44.1 every time (avoids cache)
+ARG GRADIO_FORCE_REINSTALL=1
+RUN pip install --no-cache-dir --upgrade gradio==4.44.1
+
 
-# ------------------------------------------------------
-# ✅ Install minimal system deps (needed by OpenCV & dlib)
-# ------------------------------------------------------
+# System deps for OpenCV + dlib
 RUN apt-get update && apt-get install -y --no-install-recommends \
-    libgl1 libglib2.0-0 && \
-    rm -rf /var/lib/apt/lists/*
-
-# ------------------------------------------------------
-# ✅ Install all Python dependencies (pinned + GPU safe)
-# ------------------------------------------------------
-RUN pip install --no-cache-dir \
-    tensorflow==2.18.0 \
-    torch==2.2.2 \
-    torchvision==0.17.2 \
-    facenet-pytorch==2.6.0 \
-    opencv-python \
-    imutils \
-    dlib-bin==19.24.6 \
-    gradio==4.42.0 \
-    scipy \
-    "numpy<2.0"
-
-# ------------------------------------------------------
-# ✅ Environment tuning
-# ------------------------------------------------------
-# Safe OpenMP threads (integer only)
-ENV OMP_NUM_THREADS=4
-ENV MKL_NUM_THREADS=4
-
-# Disable OneDNN auto-tuning (stabilizes TF startup)
-ENV TF_ENABLE_ONEDNN_OPTS=0
-
-# Redirect Matplotlib config to a writable temp directory
-ENV MPLCONFIGDIR=/tmp/matplotlib
-
-# Prevent Python stdout buffering (helps Gradio logs)
-ENV PYTHONUNBUFFERED=1
-
-# ------------------------------------------------------
-# ✅ Expose the Gradio port
-# ------------------------------------------------------
-EXPOSE 7860
+    build-essential cmake \
+    libgl1 libglib2.0-0 libsm6 libxext6 libxrender1 \
+    && rm -rf /var/lib/apt/lists/*
+
+WORKDIR /app
+
+# Copy app + requirements
+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/
 
-# ------------------------------------------------------
-# ✅ Launch the app
-# ------------------------------------------------------
+# Install TensorFlow (GPU-capable) + requirements
+RUN pip install --no-cache-dir tensorflow==2.18.0
+RUN pip install --no-cache-dir -r requirements.txt
+
+EXPOSE 7860
 CMD ["python", "app.py"]