transcribe-diarize / src /constants.py
Ratnesh-dev's picture
Add Pyannote Diarization API
2e9f41b
raw
history blame contribute delete
881 Bytes
WHISPER_LARGE_V3_TURBO = "Whisper Large V3 Turbo"
PARAKEET_V3 = "NVIDIA Parakeet v3"
PYANNOTE_COMMUNITY_1 = "pyannote/speaker-diarization-community-1"
SUPPORTED_MODELS = [
WHISPER_LARGE_V3_TURBO,
PARAKEET_V3,
]
OMITTED_MODELS = {
"Whisper Large V3": "Removed from this benchmark-focused app per configuration.",
"Whisper.cpp (large)": "Removed from this benchmark-focused app per configuration.",
"Whisper faster (large)": "Removed from this benchmark-focused app per configuration.",
"IBM Granite Speech 3.3 8B": (
"Omitted because a stable, documented word-level timestamp interface is not available "
"in standard inference usage."
)
}
MODEL_IDS = {
WHISPER_LARGE_V3_TURBO: "openai/whisper-large-v3-turbo",
PARAKEET_V3: "nvidia/parakeet-tdt-0.6b-v3",
PYANNOTE_COMMUNITY_1: "pyannote/speaker-diarization-community-1",
}