LAP-DEV commited on
Commit
c77b581
·
verified ·
1 Parent(s): e89cc88

Update modules/whisper/whisper_base.py

Browse files
Files changed (1) hide show
  1. modules/whisper/whisper_base.py +3 -3
modules/whisper/whisper_base.py CHANGED
@@ -265,7 +265,7 @@ class WhisperBase(ABC):
265
  # Translate to English using Whisper built-in functionality
266
  transcription_note = ""
267
  if params.is_translate:
268
- transcription_note = "Translated to English"
269
  source_lang = "English"
270
 
271
  # Translate the transcribed segments
@@ -281,7 +281,7 @@ class WhisperBase(ABC):
281
  speaker_diarization=params.is_diarize
282
  )
283
  if source_lang != target_lang:
284
- translation_note = source_lang + " to " + target_lang
285
  else:
286
  translation_note = "None (Already in " + target_lang + ")"
287
  else:
@@ -334,7 +334,7 @@ class WhisperBase(ABC):
334
  total_info += f'Input file: {info["input_source_file"]}\nLanguage: {info["lang"]} (Probability {info["lang_prob"]}%)\n'
335
 
336
  if params.is_translate:
337
- total_info += f'Transcription: {info["transcription"]}\n'
338
 
339
  if translate_output:
340
  total_info += f'Translation: {info["translation"]}\n'
 
265
  # Translate to English using Whisper built-in functionality
266
  transcription_note = ""
267
  if params.is_translate:
268
+ transcription_note = "To English (using Whisper)"
269
  source_lang = "English"
270
 
271
  # Translate the transcribed segments
 
281
  speaker_diarization=params.is_diarize
282
  )
283
  if source_lang != target_lang:
284
+ translation_note = source_lang + " to " + target_lang + " (using Facebook NLLB)"
285
  else:
286
  translation_note = "None (Already in " + target_lang + ")"
287
  else:
 
334
  total_info += f'Input file: {info["input_source_file"]}\nLanguage: {info["lang"]} (Probability {info["lang_prob"]}%)\n'
335
 
336
  if params.is_translate:
337
+ total_info += f'Translation: {info["transcription"]}\n'
338
 
339
  if translate_output:
340
  total_info += f'Translation: {info["translation"]}\n'