Update modules/whisper/whisper_base.py

modules/whisper/whisper_base.py

@@ -268,14 +268,14 @@ 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
-                self.nllb_inf = NLLBInference()
-                test = self.nllb_inf.translate_text(
-                    input_list_dict=transcribed_segments,
-                    model_size="facebook/nllb-200-distilled-600M",
-                    src_lang=file_language,
-                    tgt_lang="Dutch"
-                )  
-                print(test)
+                if translate_output:
+                    self.nllb_inf = NLLBInference()
+                    transcribed_segments = self.nllb_inf.translate_text(
+                        input_list_dict=transcribed_segments,
+                        model_size=translate_model,
+                        src_lang=file_language,
+                        tgt_lang=target_lang
+                    )
                 
                 ## Get preview as txt
                 file_name, file_ext = os.path.splitext(os.path.basename(file))