services:
  app:
    build: .
    ports:
      - "8000:8000"
    volumes:
      # Persist HuggingFace model cache between restarts
      - hf-cache:/data/huggingface
      # Persist engine state and trained models
      - engine-state:/data/engine_state
      - ./trained_model:/data/trained_model
    environment:
      - HOST=0.0.0.0
      - PORT=8000

volumes:
  hf-cache:
  engine-state: