version: '3.9'

services:
  imageserver:
    image: imageserver
    build: ./
    runtime: nvidia
    ports:
      - 8000:8000
    env_file:
      - .env
    volumes:
      - ./models:/root/.cache/huggingface/diffusers/
      - ./main.py:/app/main.py
      - ./images:/images
    restart: unless-stopped