docker-compose.gpu.yml

services:
  reaspeech-gpu:
    build:
      context: .
      dockerfile: Dockerfile.gpu
    entrypoint: ["python3", "app/run.py", "--build-reascripts"]
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: 1
              capabilities: [gpu]
    environment:
      - ASR_MODEL=small
      - ASR_ENGINE=faster_whisper
    ports:
      - "9000:9000"
    restart: on-failure
    volumes:
      - ./app:/app/app
      - ./reascripts:/app/reascripts
      - cache-pip:/root/.cache/pip
      - cache-whisper:/root/.cache/whisper

volumes:
  cache-pip:
  cache-whisper: