Update modules/whisper/whisper_base.py
Browse files
modules/whisper/whisper_base.py
CHANGED
@@ -338,22 +338,14 @@ 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:\t{info["transcription"]}\n\t
|
342 |
|
343 |
if translate_output:
|
344 |
-
total_info += f'Translation:\t{info["translation"]}\n\t
|
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 |
-
if params.is_translate and not translate_output:
|
351 |
-
total_info += "\n\nNote: translation handled by OpenAI Whisper"
|
352 |
-
if not params.is_translate and translate_output:
|
353 |
-
total_info += "\n\nNote: translation handled by Facebook NLLB"
|
354 |
-
if params.is_translate and translate_output:
|
355 |
-
total_info += "\n\nNote: translation handled by OpenAI Whisper & Facebook NLLB"
|
356 |
-
|
357 |
result_str = total_result.rstrip("\n")
|
358 |
result_file_path = [info['path'] for info in files_to_download.values()]
|
359 |
|
|
|
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\t⤷ handled by OpenAI Whisper\n'
|
342 |
|
343 |
if translate_output:
|
344 |
+
total_info += f'Translation:\t{info["translation"]}\n\t⤷ handled by Facebook NLLB\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 |
result_str = total_result.rstrip("\n")
|
350 |
result_file_path = [info['path'] for info in files_to_download.values()]
|
351 |
|