docker-compose.yml

version: "3.9"

services:
  image-gen:
    build: .
    ports:
      - "8000:8000"
    deploy:
      resources:
        reservations:
          devices:
            - capabilities: [gpu]