Update modules/whisper/whisper_base.py
Browse files
modules/whisper/whisper_base.py
CHANGED
|
@@ -338,16 +338,16 @@ class WhisperBase(ABC):
|
|
| 338 |
total_info += f'Input file:\t\t{info["input_source_file"]}\nLanguage:\t{info["lang"]} (probability {info["lang_prob"]}%)\n'
|
| 339 |
|
| 340 |
if params.is_translate:
|
| 341 |
-
total_info += f'Translation
|
| 342 |
|
| 343 |
if translate_output:
|
| 344 |
-
total_info += f'Translation
|
| 345 |
|
| 346 |
time_end = datetime.now()
|
| 347 |
total_info += f"\nTotal processing time: {self.format_time((time_end-time_start).total_seconds())}"
|
| 348 |
|
| 349 |
if params.is_translate or translate_output:
|
| 350 |
-
total_info += "\n\n
|
| 351 |
|
| 352 |
result_str = total_result.rstrip("\n")
|
| 353 |
result_file_path = [info['path'] for info in files_to_download.values()]
|
|
|
|
| 338 |
total_info += f'Input file:\t\t{info["input_source_file"]}\nLanguage:\t{info["lang"]} (probability {info["lang_prob"]}%)\n'
|
| 339 |
|
| 340 |
if params.is_translate:
|
| 341 |
+
total_info += f'Translation¹:\t{info["transcription"]}\n'
|
| 342 |
|
| 343 |
if translate_output:
|
| 344 |
+
total_info += f'Translation²:\t{info["translation"]}\n'
|
| 345 |
|
| 346 |
time_end = datetime.now()
|
| 347 |
total_info += f"\nTotal processing time: {self.format_time((time_end-time_start).total_seconds())}"
|
| 348 |
|
| 349 |
if params.is_translate or translate_output:
|
| 350 |
+
total_info += "\n\n¹\tUsing OpenAI Whisper\n²\tUsing Facebook NLLB"
|
| 351 |
|
| 352 |
result_str = total_result.rstrip("\n")
|
| 353 |
result_file_path = [info['path'] for info in files_to_download.values()]
|