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
-import modules.translation.translation_base as TranslationBaseClass
+from modules.translation.translation_base import TranslationBase
 
 class WhisperBase(ABC):
     def __init__(self,
@@ -259,8 +259,8 @@ 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
-                TranslateInstance = TranslationBaseClass()
-                test = TranslateInstance.translate_text(
+                #TranslateInstance = TranslationBaseClass()
+                #test = TranslateInstance.translate_text(
                     input_list_dict=transcribed_segments,
                     model_size="facebook/nllb-200-distilled-600M",
                     src_lang=file_language,