maximfilms/pyannote_fastapi repository overview

⁠pyannote speaker diarization — FastAPI service

A minimal FastAPI HTTP wrapper around pyannote/speaker-diarization-community-1⁠. Upload an audio file and stream the diarization result back over Server-Sent Events, with periodic heartbeats so long jobs don't time out behind proxies.

Application code only — model weights are not bundled. The upstream pipeline is gated on Hugging Face under CC-BY-4.0; you must supply your own HF_TOKEN (after accepting the model card terms) or mount an offline checkout via MODEL_PATH.

⁠Tags

• latest, <version> — CUDA build, base pytorch/pytorch:*-cuda*-cudnn9-runtime (auto-selected at build time)
• latest-cpu, <version>-cpu — CPU-only build on python:3.11-slim
• sha-<short> — immutable commit builds from main

⁠Run

docker run --rm -it --gpus all \
  -e API_KEYS="replace-me" \
  -e HF_TOKEN="replace-me" \
  -v pyannote_hf_cache:/opt/huggingface \
  -p 8000:8000 \
  maximsachs/pyannote_fastapi:latest

CPU-only host: drop --gpus all and use the :latest-cpu tag.

⁠Call it

curl -N -fsS \
  -H "Authorization: Bearer replace-me" \
  -H "Accept: text/event-stream" \
  -F "file=@/path/to/audio.wav" \
  http://127.0.0.1:8000/diarize

You'll receive a text/event-stream of status and heartbeat events, ending in a result event with the speaker segments.

⁠Endpoints

⁠Key environment variables

⁠Source and full docs

github.com/maximsachs/pyannote_fastapi⁠ — full API reference, every error code, SSE event schema, and tuning guide in docs/API.md. Wrapper code is MIT; the model itself remains CC-BY-4.0 and gated.