Update modules/whisper/whisper_base.py
Browse files
modules/whisper/whisper_base.py
CHANGED
@@ -332,7 +332,7 @@ class WhisperBase(ABC):
|
|
332 |
for file_name, info in files_info.items():
|
333 |
total_result += f'{info["subtitle"]}'
|
334 |
total_time += info["time_for_task"]
|
335 |
-
total_info += f'Input file:\t\t {info["input_source_file"]}\nLanguage:\t {info["lang"]} (
|
336 |
|
337 |
if params.is_translate:
|
338 |
total_info += f'Translation:\t {info["transcription"]}\n'
|
@@ -341,7 +341,7 @@ class WhisperBase(ABC):
|
|
341 |
total_info += f'Translation:\t {info["translation"]}\n'
|
342 |
|
343 |
time_end = datetime.now()
|
344 |
-
total_info += f"\
|
345 |
|
346 |
result_str = total_result.rstrip("\n")
|
347 |
result_file_path = [info['path'] for info in files_to_download.values()]
|
|
|
332 |
for file_name, info in files_info.items():
|
333 |
total_result += f'{info["subtitle"]}'
|
334 |
total_time += info["time_for_task"]
|
335 |
+
total_info += f'Input file:\t\t {info["input_source_file"]}\nLanguage:\t {info["lang"]} (probability {info["lang_prob"]}%)\n'
|
336 |
|
337 |
if params.is_translate:
|
338 |
total_info += f'Translation:\t {info["transcription"]}\n'
|
|
|
341 |
total_info += f'Translation:\t {info["translation"]}\n'
|
342 |
|
343 |
time_end = datetime.now()
|
344 |
+
total_info += f"\nTotal processing time: {self.format_time((time_end-time_start).total_seconds())}"
|
345 |
|
346 |
result_str = total_result.rstrip("\n")
|
347 |
result_file_path = [info['path'] for info in files_to_download.values()]
|