services:
  webui:
    build:
      context: .
      args:
        - DEVICE=cpu # 'cpu' for Mac, change to 'gpu' on a Linux server with NVIDIA GPU
    image: yolo-train-webui:latest
    ports:
      - "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]
    restart: unless-stopped