services:
  ai:
    container_name: "milchchanai"
    build:
      context: .
    restart: unless-stopped
    tty: true
    env_file:
      - .env
    environment:
      GRADIO_SERVER_NAME: 0.0.0.0
      GRADIO_SERVER_PORT: 7860
      HF_HOME: /data/huggingface
      WHISPER_CACHE_DIR: /data/whisper
    volumes:
      - hf-cache:/data/huggingface
      - whisper-cache:/data/whisper
    ports:
      - "7860:7860"
    deploy:
      resources:
        reservations:
          devices:
            - capabilities: [gpu]
volumes:
  hf-cache:
  whisper-cache: