From 8074af0b526d51d8a779f97fa4276330606d1a03 Mon Sep 17 00:00:00 2001
From: Josh <1113285+jsrozner@users.noreply.github.com>
Date: Sun, 17 Mar 2024 13:56:16 -0700
Subject: [PATCH] Default to 3.1 diarizer model

Current implementation (following the readme example) gives error:
ImportError: 'speechbrain' must be installed to use 'speechbrain/spkrec-ecapa-voxceleb' embeddings. Visit https://speechbrain.github.io for installation instructions.

Using 3.1 model fixes this. This fix should result in readme example code correctly executing
---
 src/speechbox/diarize.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/speechbox/diarize.py b/src/speechbox/diarize.py
index 7bbf03e..fc5421d 100644
--- a/src/speechbox/diarize.py
+++ b/src/speechbox/diarize.py
@@ -25,7 +25,7 @@ def from_pretrained(
         cls,
         asr_model: Optional[str] = "openai/whisper-medium",
         *,
-        diarizer_model: Optional[str] = "pyannote/speaker-diarization",
+        diarizer_model: Optional[str] = "pyannote/speaker-diarization-3.1",
         chunk_length_s: Optional[int] = 30,
         use_auth_token: Optional[Union[str, bool]] = True,
         **kwargs,