Update modules/whisper/whisper_base.py

modules/whisper/whisper_base.py

@@ -258,9 +258,13 @@ class WhisperBase(ABC):
                 #)
                 #files_info[file_name] = {"subtitle": subtitle, "time_for_task": time_for_task, "path": file_path, "lang": file_language, "lang_prob": file_lang_probs, "input_source_file": (file_name+file_ext)}
 
-
                 # Translate the trandscribed segments
-                test = translate_text(transcribed_segments,"facebook/nllb-200-distilled-600M","English","Dutch")
+                test = TranslationBase.translate_file(self,
+                       input_list_dict: transcribed_segments,
+                       model_size: "facebook/nllb-200-distilled-600M",
+                       src_lang: file_language,
+                       tgt_lang: "Dutch"
+                )  
                 print(test)
                 
                 ## Get preview as txt