Merge pull request #586 from remic33/main

setup.py

@@ -1,15 +1,8 @@
 import os
 import platform
+
 import pkg_resources
-from setuptools import setup, find_packages
+from setuptools import find_packages, setup
-
-
-def get_pyannote_audio_version():
-    machine = platform.machine()
-    system = platform.system()
-    version = "3.0.0" if machine == "aarch64" or system == "Darwin" else "3.0.1"
-    return version
-
 
 setup(
     name="whisperx",
@@ -28,7 +21,7 @@ setup(
             open(os.path.join(os.path.dirname(__file__), "requirements.txt"))
         )
     ]
-    + [f"pyannote.audio=={get_pyannote_audio_version()}"],
+    + [f"pyannote.audio==3.1.0"],
     entry_points={
         "console_scripts": ["whisperx=whisperx.transcribe:cli"],
     },

whisperx/diarize.py

@@ -10,7 +10,7 @@ from .audio import load_audio, SAMPLE_RATE
 class DiarizationPipeline:
     def __init__(
         self,
-        model_name="pyannote/speaker-diarization-3.0",
+        model_name="pyannote/speaker-diarization-3.1",
         use_auth_token=None,
         device: Optional[Union[str, torch.device]] = "cpu",
     ):