Update modules/whisper/whisper_base.py

modules/whisper/whisper_base.py

@@ -245,6 +245,7 @@ class WhisperBase(ABC):
                         translate_model: str = "",
                         target_lang: str = "",
                         add_timestamp_preview: bool = False,
+                        diarize_speakers: bool = False,
                         progress=gr.Progress(),
                         *whisper_params,
                         ) -> list:
@@ -276,6 +277,8 @@ class WhisperBase(ABC):
             Target language to use
         add_timestamp_preview: bool
             Boolean value from gr.Checkbox() that determines whether to add a timestamp to output preview
+        diarize_speakers: bool
+            Boolean value from gr.Checkbox() that determines whether to diarize speakers
         progress: gr.Progress
             Indicator to show progress directly in gradio.
         *whisper_params: tuple