services:
  webui:
    build:
      context: .
    image: yolo-train-webui:latest
    network_mode: bridge
    ports:
      - "8000:8000"
    volumes:
      - ./datasets:/workspace/datasets
      - ./runs:/workspace/runs
      - ./models:/workspace/models
      - ./models/.config:/root/.config/Ultralytics
    environment:
      # Tracking URI for external MLflow server. Change host/IP if running MLflow on another machine/container.
      YOLO_WEBUI_MLFLOW_TRACKING_URI: http://host.docker.internal:5000
    extra_hosts:
      - "host.docker.internal:host-gateway"
    # 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