Update modules/whisper/whisper_base.py
Browse files
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 = "
|
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'
|
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'
|