From 9e3145ceadc6f9826a9374334c647fc3ce92b093 Mon Sep 17 00:00:00 2001 From: Dudu Asulin <46293514+davidas1@users.noreply.github.com> Date: Wed, 2 Aug 2023 10:36:56 +0300 Subject: [PATCH] more --- whisperx/diarize.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/whisperx/diarize.py b/whisperx/diarize.py index eae6a19..e50dc0f 100644 --- a/whisperx/diarize.py +++ b/whisperx/diarize.py @@ -17,7 +17,7 @@ class DiarizationPipeline: device = torch.device(device) self.model = Pipeline.from_pretrained(model_name, use_auth_token=use_auth_token).to(device) - def __call__(self, audio, min_speakers=None, max_speakers=None): + def __call__(self, audio: Union[str, np.ndarray], min_speakers=None, max_speakers=None): if isinstance(audio, str): audio = load_audio(audio) audio_data = {