Update Dockerfile

Dockerfile

@@ -1,36 +1,42 @@
-# ============================================================
-# GPU-ENABLED SELF-CONTAINED IMAGE FOR DEEPFAKE VIDEO DETECTOR
-# ============================================================
-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)
-ARG GRADIO_FORCE_REINSTALL=1
-RUN pip install --no-cache-dir --upgrade gradio==4.44.1
-
-
-# System deps for OpenCV + dlib
-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 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
-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"]
+# ============================================================
+# Deepfake Video Detector - Hugging Face GPU Dockerfile
+# ============================================================
+# Use Hugging Face's official GPU image (includes CUDA + PyTorch)
+FROM ghcr.io/huggingface/transformers-pytorch-gpu:latest
+
+# Disable prompts and enable noninteractive mode
+ENV DEBIAN_FRONTEND=noninteractive \
+    PYTHONDONTWRITEBYTECODE=1 \
+    PYTHONUNBUFFERED=1
+
+# Install system dependencies
+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 project files into the image
+# ------------------------------------------------------------
+COPY requirements.txt /app/requirements.txt
+COPY app.py /app/app.py
+COPY models /app/models
+COPY shape_predictor_68_face_landmarks.dat /app/
+
+# ------------------------------------------------------------
+# Install dependencies
+# ------------------------------------------------------------
+RUN pip install --no-cache-dir tensorflow==2.18.0
+RUN pip install --no-cache-dir -r requirements.txt
+
+# Force-remove old Gradio and install the latest stable version
+RUN pip uninstall -y gradio gradio_client || true
+RUN pip install --no-cache-dir --upgrade gradio==4.44.1 gradio_client==0.16.0
+
+# ------------------------------------------------------------
+# Expose app port and launch
+# ------------------------------------------------------------
+EXPOSE 7860
+CMD ["python", "app.py"]