LAP-DEV commited on
Commit
1e819f6
·
verified ·
1 Parent(s): efa53b5

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
@@ -270,6 +270,7 @@ class WhisperBase(ABC):
270
  #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)}
271
 
272
  # Translate the transcribed segments
 
273
  if translate_output:
274
  self.nllb_inf = NLLBInference()
275
  if file_language in NLLB_AVAILABLE_LANGS.keys():
@@ -280,7 +281,7 @@ class WhisperBase(ABC):
280
  tgt_lang=target_lang,
281
  speaker_diarization=params.is_diarize
282
  )
283
- translation_note = "from " + file_language + " to " + target_lang
284
  else:
285
  translation_note = file_language + " not supported"
286
 
 
270
  #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)}
271
 
272
  # Translate the transcribed segments
273
+ translation_note = ""
274
  if translate_output:
275
  self.nllb_inf = NLLBInference()
276
  if file_language in NLLB_AVAILABLE_LANGS.keys():
 
281
  tgt_lang=target_lang,
282
  speaker_diarization=params.is_diarize
283
  )
284
+ translation_note = file_language + " to " + target_lang
285
  else:
286
  translation_note = file_language + " not supported"
287