Update modules/translation/translation_base.py

modules/translation/translation_base.py

@@ -142,6 +142,7 @@ class TranslationBase(ABC):
                        model_size: str,
                        src_lang: str,
                        tgt_lang: str,
+                       speaker_diarization: bool = False,
                        max_length: int = 200,
                        add_timestamp: bool = True,
                        progress=gr.Progress()) -> list:
@@ -157,6 +158,8 @@ class TranslationBase(ABC):
             Source language of the file to translate from gr.Dropdown()
         tgt_lang: str
             Target language of the file to translate from gr.Dropdown()
+        speaker_diarization: bool
+            Boolean value that determines whether diarization is enabled or not
         max_length: int
             Max length per line to translate
         add_timestamp: bool