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.translation_base import translate_text
+from modules.translation.translation_base import *
 
 class WhisperBase(ABC):
     def __init__(self,
@@ -260,8 +260,7 @@ class WhisperBase(ABC):
 
 
                 # Translate the trandscribed segments
-                # Translate the trandscribed segments
-                test=translate_text(transcribed_segments,"facebook/nllb-200-1.3B",file_language,"Dutch") 
+                test = self.translate_text(transcribed_segments,"facebook/nllb-200-1.3B",file_language,"Dutch") 
                 print(test)
                 
                 ## Get preview as txt