Update modules/whisper/whisper_base.py
Browse files
modules/whisper/whisper_base.py
CHANGED
@@ -266,8 +266,11 @@ class WhisperBase(ABC):
|
|
266 |
# Translate to English using Whisper built-in functionality
|
267 |
transcription_note = ""
|
268 |
if params.is_translate:
|
269 |
-
|
270 |
-
|
|
|
|
|
|
|
271 |
|
272 |
# Translate the transcribed segments
|
273 |
translation_note = ""
|
@@ -282,9 +285,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 |
+
if source_lang != "English":
|
270 |
+
transcription_note = "To English"
|
271 |
+
source_lang = "English"
|
272 |
+
else:
|
273 |
+
transcription_note = "Already in " + target_lang
|
274 |
|
275 |
# Translate the transcribed segments
|
276 |
translation_note = ""
|
|
|
285 |
tgt_lang=target_lang,
|
286 |
speaker_diarization=params.is_diarize
|
287 |
)
|
288 |
+
translation_note = "To " + target_lang
|
289 |
else:
|
290 |
+
translation_note = source_lang + " not supported"
|
291 |
else:
|
292 |
translation_note = "Already in " + target_lang
|
293 |
|