version: '3.8'

services:
  wav2vec-service:
    image: duyta506/graduation:wav2vec-lastest
    container_name: wav2vec-container
    ports:
      - "3000:3000"
    restart: unless-stopped
    environment:
      - NVIDIA_VISIBLE_DEVICES=all
      - NVIDIA_DRIVER_CAPABILITIES=compute,utility
    deploy:
      resources:
        reservations:
          devices:
            - capabilities: [gpu]
    runtime: nvidia

  whisper-service:
    image: duyta506/graduation:whisper-latest-gpu12.0
    container_name: whisper-container
    ports:
      - "5678:5678"
    restart: unless-stopped
    environment:
      - NVIDIA_VISIBLE_DEVICES=all
      - NVIDIA_DRIVER_CAPABILITIES=compute,utility
    deploy:
      resources:
        reservations:
          devices:
            - capabilities: [gpu]
    runtime: nvidia