Spaces:

HighnessAnnotation
/

ABAB-Annotation

Paused

Vishal Patil commited on Mar 10

Commit

d367403

verified ·

1 Parent(s): ad34846

Update pipeline.py

Files changed (1) hide show

pipeline.py CHANGED Viewed

@@ -78,7 +78,7 @@ class SpeechAnnotationPipeline:
             logger.info("Loading pyannote diarization pipeline...")
             self._diarization_pipeline = PyannotePipeline.from_pretrained(
                 "pyannote/speaker-diarization-3.1",
-                use_auth_token=HF_TOKEN,
             )
             if DEVICE == "cuda":
                 self._diarization_pipeline = self._diarization_pipeline.to(torch.device("cuda"))

             logger.info("Loading pyannote diarization pipeline...")
             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"))