LAP-DEV commited on
Commit
d0a0071
·
verified ·
1 Parent(s): 3f37fef

Update modules/whisper/whisper_base.py

Browse files
Files changed (1) hide show
  1. modules/whisper/whisper_base.py +9 -5
modules/whisper/whisper_base.py CHANGED
@@ -376,16 +376,20 @@ class WhisperBase(ABC):
376
  total_result += f'{info["subtitle"]}'
377
 
378
  total_time += info["time_for_task"]
379
- total_info += f'Input file:\t{info["input_source_file"]}\nLanguage:\t{info["lang"]} (probability {info["lang_prob"]}%)\n'
380
 
381
- total_info += ("Input file:").ljust(12)+"\t"+str(info["input_source_file"])+"\n"
382
- total_info += ("Language:").ljust(12)+"\t"+str(info["lang"])+"(probability "+str(info["lang_prob"])+"%)\n"
383
 
384
  if params.is_translate:
385
- total_info += f'Translation:\t{info["transcription"]}\n\t⤷ Handled by OpenAI Whisper\n'
 
 
386
 
387
  if translate_output:
388
- total_info += f'Translation:\t{info["translation"]}\n\t⤷ Handled by Facebook NLLB\n'
 
 
389
 
390
  time_end = datetime.now()
391
  total_info += f"\nTotal processing time: {self.format_time((time_end-time_start).total_seconds())}"
 
376
  total_result += f'{info["subtitle"]}'
377
 
378
  total_time += info["time_for_task"]
379
+ #total_info += f'Input file:\t{info["input_source_file"]}\nLanguage:\t{info["lang"]} (probability {info["lang_prob"]}%)\n'
380
 
381
+ total_info += ("Input file:").ljust(15)+str(info["input_source_file"])+"\n"
382
+ total_info += ("Language:").ljust(15)+str(info["lang"])+"(probability "+str(info["lang_prob"])+"%)\n"
383
 
384
  if params.is_translate:
385
+ #total_info += f'Translation:\t{info["transcription"]}\n\t⤷ Handled by OpenAI Whisper\n'
386
+ total_info += "Translation:"+str(info["transcription"])+"\n"
387
+ total_info += ("").ljust(8)+"⤷ Handled by OpenAI Whisper\n"
388
 
389
  if translate_output:
390
+ #total_info += f'Translation:\t{info["translation"]}\n\t⤷ Handled by Facebook NLLB\n'
391
+ total_info += "Translation:"+str(info["translation"])+"\n"
392
+ total_info += ("").ljust(8)+"⤷ Handled by Facebook NLLB\n"
393
 
394
  time_end = datetime.now()
395
  total_info += f"\nTotal processing time: {self.format_time((time_end-time_start).total_seconds())}"