DeepFake-Detector/docker-compose.yml at main · BhavyaKansal20/DeepFake-Detector · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
version: "3.9"

services:
  # ─── Main App (API + UI) ───────────────────────────────────────────────────
  deepfake-detector:
    build: .
    container_name: deepfake-detector
    ports:
      - "8000:8000"    # FastAPI
      - "8501:8501"    # Streamlit
    volumes:
      - ./models:/app/models          # persist trained models
      - ./data:/app/data              # datasets
      - ./logs:/app/logs              # logs
    environment:
      - CONFIG_PATH=/app/configs/config.yaml
      - MODELS_DIR=/app/models
      - CUDA_VISIBLE_DEVICES=0
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: 1
              capabilities: [gpu]
    restart: unless-stopped
    healthcheck:
      test: ["CMD", "curl", "-f", "http://localhost:8000/health"]
      interval: 30s
      timeout: 10s
      retries: 3

  # ─── Optional: Nginx reverse proxy ────────────────────────────────────────
  nginx:
    image: nginx:alpine
    container_name: deepfake-nginx
    ports:
      - "80:80"
      - "443:443"
    volumes:
      - ./configs/nginx.conf:/etc/nginx/conf.d/default.conf:ro
    depends_on:
      - deepfake-detector
    restart: unless-stopped
    profiles:
      - production   # only starts with: docker-compose --profile production up

volumes:
  models_data:
  logs_data: