Update asr_diarizer.py

asr_diarizer.py

@@ -9,19 +9,30 @@ from transformers import pipeline
 class ASRDiarizationPipeline:
     def __init__(
         self,
+        asr_pipeline,
+        diarization_pipeline,
+    ):
+        self.asr_pipeline = asr_pipeline
+        self.diarization_pipeline = diarization_pipeline
+
+    @classmethod
+    def from_pretrained(
+        cls,
         asr_model: Optional[str] = "openai/whisper-small",
         diarizer_model: Optional[str] = "pyannote/speaker-diarization",
         chunk_length_s: int = 30,
+        use_auth_token: Union[str, bool] = True,
         **kwargs,
     ):
-        self.asr_pipeline = pipeline(
+        asr_pipeline = pipeline(
             "automatic-speech-recognition",
             model=asr_model,
-            use_auth_token=True,
             chunk_length_s=chunk_length_s,
+            use_auth_token=use_auth_token,
             **kwargs,
         )
-        self.diarization_pipeline = Pipeline.from_pretrained(diarizer_model, use_auth_token=True)
+        diarization_pipeline = Pipeline.from_pretrained(diarizer_model, use_auth_token=use_auth_token)
+        cls(asr_pipeline, diarization_pipeline)
 
     def __call__(
         self,