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 |
|