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