version: "3.7"

services:
  pyannote_speaker_diarization:
    image: pyannote-speaker-diarization:v0.1
    environment:
      - GIT_PYTHON_REFRESH=quiet
      - HUGGINGFACE_HUB_CACHE=/home/user/temp/.cache
      - TRANSFORMERS_CACHE=/home/user/temp/.cache
      - HF_HOME=/home/user/temp/.cache
      - TORCH_HOME=/home/user/temp/torch
      - TRANSFORMERS_OFFLINE=1
    tty: true
    stdin_open: true
    volumes:
      - ../torch/:/home/user/temp/torch
    build:
      context: .
    container_name: pyannote-speaker-diarization
    ports:
      - 8007:9000