LAP-DEV commited on
Commit
a4c3b4f
·
verified ·
1 Parent(s): 1b23226

Update modules/whisper/whisper_base.py

Browse files
Files changed (1) hide show
  1. modules/whisper/whisper_base.py +2 -10
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\t\t(handled by OpenAI Whisper)\n'
342
 
343
  if translate_output:
344
- total_info += f'Translation:\t{info["translation"]}\n\t\t\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
- 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\thandled by OpenAI Whisper\n'
342
 
343
  if translate_output:
344
+ total_info += f'Translation:\t{info["translation"]}\n\thandled 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