Update pipeline.py

pipeline.py

@@ -1,7 +1,7 @@
 """
 Speech Annotation Pipeline
 - Speaker Diarization via pyannote.audio
-- Transcription via openai-whisper (CPU-friendly, no av dependency)
+- Transcription via openai-whisper
 """
 
 import os
@@ -72,14 +72,22 @@ class SpeechAnnotationPipeline:
         if self._diarization_pipeline is None:
             if not HF_TOKEN:
                 raise EnvironmentError(
-                    "HF_TOKEN environment variable is required. "
-                    "Set it in Space Settings > Variables and secrets."
+                    "HF_TOKEN not set. Add it in Space Settings > Variables and secrets."
                 )
             logger.info("Loading pyannote diarization pipeline...")
-            self._diarization_pipeline = PyannotePipeline.from_pretrained(
-                "pyannote/speaker-diarization-3.1",
-                token=HF_TOKEN,
-            )
+
+            # Try both parameter names for compatibility
+            try:
+                self._diarization_pipeline = PyannotePipeline.from_pretrained(
+                    "pyannote/speaker-diarization-3.1",
+                    use_auth_token=HF_TOKEN,
+                )
+            except TypeError:
+                self._diarization_pipeline = PyannotePipeline.from_pretrained(
+                    "pyannote/speaker-diarization-3.1",
+                    token=HF_TOKEN,
+                )
+
             if DEVICE == "cuda":
                 self._diarization_pipeline = self._diarization_pipeline.to(torch.device("cuda"))
         return self._diarization_pipeline