Update modules/whisper/whisper_base.py

modules/whisper/whisper_base.py

@@ -179,6 +179,9 @@ class WhisperBase(ABC):
                         input_folder_path: Optional[str] = None,
                         file_format: str = "SRT",
                         add_timestamp: bool = True,
+                        translate_output: bool = False,
+                        dd_translate_model: str = "",
+                        dd_target_lang: str = "",
                         progress=gr.Progress(),
                         *whisper_params,
                         ) -> list:
@@ -196,6 +199,12 @@ class WhisperBase(ABC):
             Subtitle File format to write from gr.Dropdown(). Supported format: [SRT, WebVTT, txt]
         add_timestamp: bool
             Boolean value from gr.Checkbox() that determines whether to add a timestamp at the end of the subtitle filename.
+        translate_output: bool
+            Translate ouput
+        translate_model: str
+            Translation model to use
+        target_lang: str
+            Target language to use
         progress: gr.Progress
             Indicator to show progress directly in gradio.
         *whisper_params: tuple