Update modules/whisper/whisper_base.py

modules/whisper/whisper_base.py

@@ -19,7 +19,7 @@ from modules.utils.files_manager import get_media_files, format_gradio_files, lo
 from modules.whisper.whisper_parameter import *
 from modules.diarize.diarizer import Diarizer
 from modules.vad.silero_vad import SileroVAD
-from modules.translation.nllb_inference import NLLBInference
+from modules.translation.translation_base import translate_text
 
 class WhisperBase(ABC):
     def __init__(self,
@@ -260,13 +260,9 @@ class WhisperBase(ABC):
 
 
                 # Translate the trandscribed segments
-                translate_input = ""
-                for item in transcribed_segments:
-                    translate_input += item['text'] + "\n"	
-
-                translate_input = translate_input.rstrip()
-                translated_text = NLLBInference.translate(translate_input,200)
-                print(translated_text)
+                # Translate the trandscribed segments
+                test=translate_text(transcribed_segments,"facebook/nllb-200-1.3B",file_language,"Dutch") 
+                print(test)
                 
                 ## Get preview as txt
                 file_name, file_ext = os.path.splitext(os.path.basename(file))