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