LAP-DEV commited on
Commit
128fbf8
·
verified ·
1 Parent(s): c3b8622

Update modules/whisper/whisper_base.py

Browse files
Files changed (1) hide show
  1. modules/whisper/whisper_base.py +2 -1
modules/whisper/whisper_base.py CHANGED
@@ -259,7 +259,8 @@ class WhisperBase(ABC):
259
  #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)}
260
 
261
  # Translate the trandscribed segments
262
- test = TranslationModule.translate_text(
 
263
  input_list_dict=transcribed_segments,
264
  model_size="facebook/nllb-200-distilled-600M",
265
  src_lang=file_language,
 
259
  #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)}
260
 
261
  # Translate the trandscribed segments
262
+ TranslateInstance = TranslationBase()
263
+ test = TranslateInstance.translate_text(
264
  input_list_dict=transcribed_segments,
265
  model_size="facebook/nllb-200-distilled-600M",
266
  src_lang=file_language,