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