Update modules/whisper/whisper_base.py
Browse files
modules/whisper/whisper_base.py
CHANGED
@@ -413,10 +413,10 @@ class WhisperBase(ABC):
|
|
413 |
title_line = ""
|
414 |
if add_timestamp_preview: title_line += "□"
|
415 |
if diarize_speakers: title_line += "□"
|
416 |
-
title_line += f'Media file:\t{info["input_source_file"]}
|
417 |
-
title_line += f'
|
418 |
-
if params.is_translate: title_line += f'
|
419 |
-
if translate_output: title_line += f'
|
420 |
title_line += "■"
|
421 |
|
422 |
temp_subtitle = info["subtitle"]
|
@@ -427,12 +427,8 @@ class WhisperBase(ABC):
|
|
427 |
total_time += info["time_for_task"]
|
428 |
|
429 |
time_end = datetime.now()
|
430 |
-
#total_info += f"
|
431 |
-
|
432 |
-
temp_file_count_text = "file"
|
433 |
-
if file_count!=1:
|
434 |
-
temp_file_count_text += "s"
|
435 |
-
total_info = f"\nProcessed {file_count} {'file' if file_count == 1 else 'files'} in {self.format_time((time_end-time_start).total_seconds())}"
|
436 |
|
437 |
result_str = total_result.rstrip("\n")
|
438 |
result_str = self.transform_text_to_list(result_str,"□","■")
|
|
|
413 |
title_line = ""
|
414 |
if add_timestamp_preview: title_line += "□"
|
415 |
if diarize_speakers: title_line += "□"
|
416 |
+
title_line += f'Media file:\t{info["input_source_file"]}'
|
417 |
+
title_line += f'\nLanguage:\t{info["lang"]} (probability {info["lang_prob"]}%)'
|
418 |
+
if params.is_translate: title_line += f'\nTranslation:\t{info["transcription"]} (Handled by OpenAI Whisper)'
|
419 |
+
if translate_output: title_line += f'\nTranslation:\t{info["translation"]} (Handled by Facebook NLLB)'
|
420 |
title_line += "■"
|
421 |
|
422 |
temp_subtitle = info["subtitle"]
|
|
|
427 |
total_time += info["time_for_task"]
|
428 |
|
429 |
time_end = datetime.now()
|
430 |
+
#total_info += f"Total processing time:\t{self.format_time((time_end-time_start).total_seconds())}"
|
431 |
+
total_info = f"Processed {file_count} {'file' if file_count == 1 else 'files'} in {self.format_time((time_end-time_start).total_seconds())}"
|
|
|
|
|
|
|
|
|
432 |
|
433 |
result_str = total_result.rstrip("\n")
|
434 |
result_str = self.transform_text_to_list(result_str,"□","■")
|