services:
  webui:
    build:
      context: .
    image: yolo-train-webui:latest
    ports:
      # The training API has no built-in user accounts, so expose it locally only.
      - "0.0.0.0:8000:8000"
    volumes:
      - ./datasets:/workspace/datasets
      - ./runs:/workspace/runs
      - ./models:/workspace/models
      - ./models/.config:/root/.config/Ultralytics
    # Uncomment the block below on Linux with NVIDIA GPU to pass the graphics card into the container:
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: all
              capabilities: [gpu]
    ipc: host
    restart: unless-stopped