Deploy DeepFake Detector API - 2026-04-21 02:45:09

Dockerfile

@@ -17,12 +17,6 @@ ENV PYTHONDONTWRITEBYTECODE=1 \
     HF_HOME=/app/.hf_cache/hf_home \
     HF_XET_CACHE=/app/.hf_cache/xet
 
-# Install system dependencies
-RUN apt-get update && apt-get install -y --no-install-recommends \
-    curl \
-    git \
-    && rm -rf /var/lib/apt/lists/*
-
 # Create non-root user (HF Spaces requirement)
 RUN useradd -m -u 1000 user
 USER user
@@ -30,9 +24,14 @@ USER user
 # Set PATH for user-installed packages
 ENV PATH="/home/user/.local/bin:$PATH"
 
-# Copy requirements and install dependencies as user
-COPY --chown=user:user requirements.txt .
-RUN pip install --no-cache-dir --upgrade -r requirements.txt
+# Install runtime dependencies by default (inference + explainability + LLM).
+# Optional dependencies are local test tools.
+ARG INSTALL_OPTIONAL_DEPS=false
+COPY --chown=user:user requirements-runtime.txt requirements-optional.txt requirements.txt ./
+RUN pip install --no-cache-dir --upgrade -r requirements-runtime.txt \
+        && if [ "$INSTALL_OPTIONAL_DEPS" = "true" ]; then \
+                 pip install --no-cache-dir --upgrade -r requirements-optional.txt; \
+             fi
 
 # Copy only files required for model prefetch first.
 COPY --chown=user:user app /app/app
@@ -47,17 +46,6 @@ RUN mkdir -p /app/.hf_cache /app/.hf_cache/hf_home /app/.hf_cache/xet \
 # Switch back to user
 USER user
 
-# Prefetch model artifacts at build time so startup does not wait on model downloads.
-RUN HF_CACHE_DIR=/app/.hf_cache HF_HUB_CACHE=/app/.hf_cache HF_HOME=/app/.hf_cache/hf_home HF_XET_CACHE=/app/.hf_cache/xet python -m app.scripts.prefetch_models
-
-# Copy full project contents after prefetch so docs/tests edits do not invalidate prefetch layers.
-COPY --chown=user:user . /app
-
-# Final copy can strip executable mode from scripts on Windows hosts; re-apply execute bit.
-USER root
-RUN chmod +x /app/start.sh
-USER user
-
 # Expose default app port
 EXPOSE 7860
 

README.md

@@ -73,6 +73,7 @@ HF Spaces deploy variables (used by [backend/deploy-to-hf.sh](deploy-to-hf.sh)):
 - `HF_SPACE_WEB_URL`
 - `HF_SPACE_APP_URL`
 - `HF_DEPLOY_DIR`
+- `INSTALL_OPTIONAL_DEPS` (`true`/`false`, default `false`)
 - `HF_BUCKET_SYNC_ON_DEPLOY` (default: `true`)
 
 ## API Endpoints
@@ -99,6 +100,18 @@ docker build -t deepfake-detector-api .
 docker run -p 7860:7860 deepfake-detector-api
 ```
 
+Image defaults:
+
+- Installs `requirements-runtime.txt` (inference + explainability + LLM)
+- Skips build-time model prefetch (first boot fills `/data` cache)
+- `POST /predict` defaults to `explain=true`
+
+Enable optional test dependencies at build time:
+
+```bash
+docker build --build-arg INSTALL_OPTIONAL_DEPS=true -t deepfake-detector-api .
+```
+
 ## Deploy to Hugging Face Spaces
 
 Recommended path is the Bash deploy script.

requirements-optional.txt

@@ -0,0 +1,6 @@
+# DeepFake Detector Backend - Optional Dependencies
+# Install these for local development/testing only.
+
+# Testing
+pytest>=7.4.0,<9.0.0
+pytest-asyncio>=0.23.0,<1.0.0

requirements-runtime.txt

@@ -0,0 +1,32 @@
+# DeepFake Detector Backend - Runtime Dependencies (inference + API)
+
+# Web framework
+fastapi>=0.109.0,<1.0.0
+uvicorn[standard]>=0.27.0,<1.0.0
+
+# Data validation
+pydantic>=2.5.0,<3.0.0
+pydantic-settings>=2.1.0,<3.0.0
+
+# Core image handling
+pillow>=10.2.0,<11.0.0
+matplotlib>=3.7.0,<4.0.0
+
+# Deep Learning
+torch>=2.0.0,<3.0.0
+torchvision>=0.15.0,<1.0.0
+timm>=0.9.0
+
+# Machine Learning (for fusion models)
+scikit-learn==1.6.1
+numpy>=1.24.0,<2.0.0
+
+# Hugging Face Hub
+huggingface_hub>=0.20.0,<1.0.0
+hf_xet
+
+# Google Gemini for LLM explanations
+google-genai>=1.0.0
+
+# File uploads
+python-multipart>=0.0.6,<1.0.0

requirements.txt

@@ -1,40 +1,4 @@
-# DeepFake Detector Backend - Python Dependencies
-
-# Web framework
-fastapi>=0.109.0,<1.0.0
-uvicorn[standard]>=0.27.0,<1.0.0
-
-# Data validation
-pydantic>=2.5.0,<3.0.0
-pydantic-settings>=2.1.0,<3.0.0
-
-# Image processing
-pillow>=10.2.0,<11.0.0
-matplotlib>=3.7.0,<4.0.0
-opencv-python>=4.8.0,<5.0.0
-
-# Deep Learning
-torch>=2.0.0,<3.0.0
-torchvision>=0.15.0,<1.0.0
-timm>=0.9.0
-
-# Machine Learning (for fusion models)
-scikit-learn==1.6.1
-numpy>=1.24.0,<2.0.0
-
-# Hugging Face Hub
-huggingface_hub>=0.20.0,<1.0.0
-hf_xet
-
-# HTTP client (optional, for testing)
-httpx>=0.26.0,<1.0.0
-
-# File uploads
-python-multipart>=0.0.6,<1.0.0
-
-# Testing
-pytest>=7.4.0,<9.0.0
-pytest-asyncio>=0.23.0,<1.0.0
-
-# Google Gemini for LLM explanations
-google-genai>=1.0.0
+# Full dependency set for local development.
+# Production images should install requirements-runtime.txt only.
+-r requirements-runtime.txt
+-r requirements-optional.txt