Update modules/whisper/whisper_base.py
Browse files
modules/whisper/whisper_base.py
CHANGED
|
@@ -414,14 +414,13 @@ class WhisperBase(ABC):
|
|
| 414 |
total_info += f'\n'
|
| 415 |
|
| 416 |
# Add filename & info as first line
|
| 417 |
-
add_tabs = 0
|
| 418 |
title_line = ""
|
| 419 |
-
if add_timestamp_preview:
|
| 420 |
-
if diarize_speakers:
|
| 421 |
-
title_line
|
| 422 |
-
title_line += f'Language
|
| 423 |
-
if params.is_translate: title_line +=
|
| 424 |
-
if translate_output: title_line +=
|
| 425 |
title_line += "■"
|
| 426 |
|
| 427 |
temp_subtitle = info["subtitle"]
|
|
|
|
| 414 |
total_info += f'\n'
|
| 415 |
|
| 416 |
# Add filename & info as first line
|
|
|
|
| 417 |
title_line = ""
|
| 418 |
+
if add_timestamp_preview: title_line += "□"
|
| 419 |
+
if diarize_speakers: title_line += "□"
|
| 420 |
+
title_line += f'Media file:\t{info["input_source_file"]}\n'
|
| 421 |
+
title_line += f'Language:\t{info["lang"]} (probability {info["lang_prob"]}%)\n'
|
| 422 |
+
if params.is_translate: title_line += f'Translation:\t{info["transcription"]} (Handled by OpenAI Whisper)\n'
|
| 423 |
+
if translate_output: title_line += f'Translation:\t{info["translation"]} (Handled by Facebook NLLB)\n'
|
| 424 |
title_line += "■"
|
| 425 |
|
| 426 |
temp_subtitle = info["subtitle"]
|