Update modules/whisper/whisper_base.py
Browse files
modules/whisper/whisper_base.py
CHANGED
@@ -266,7 +266,7 @@ class WhisperBase(ABC):
|
|
266 |
# Translate to English using Whisper built-in functionality
|
267 |
transcription_note = ""
|
268 |
if params.is_translate:
|
269 |
-
transcription_note = "To English (
|
270 |
source_lang = "English"
|
271 |
|
272 |
# Translate the transcribed segments
|
@@ -282,9 +282,9 @@ class WhisperBase(ABC):
|
|
282 |
tgt_lang=target_lang,
|
283 |
speaker_diarization=params.is_diarize
|
284 |
)
|
285 |
-
translation_note = "To " + target_lang + " (
|
286 |
else:
|
287 |
-
translation_note = source_lang + " not supported (
|
288 |
else:
|
289 |
translation_note = "Already in " + target_lang
|
290 |
|
|
|
266 |
# Translate to English using Whisper built-in functionality
|
267 |
transcription_note = ""
|
268 |
if params.is_translate:
|
269 |
+
transcription_note = "To English (Whisper)"
|
270 |
source_lang = "English"
|
271 |
|
272 |
# Translate the transcribed segments
|
|
|
282 |
tgt_lang=target_lang,
|
283 |
speaker_diarization=params.is_diarize
|
284 |
)
|
285 |
+
translation_note = "To " + target_lang + " (Facebook NLLB)"
|
286 |
else:
|
287 |
+
translation_note = source_lang + " not supported (Facebook NLLB)"
|
288 |
else:
|
289 |
translation_note = "Already in " + target_lang
|
290 |
|