LAP-DEV commited on
Commit
6d2f230
·
verified ·
1 Parent(s): dc6b2c7

Update modules/whisper/whisper_base.py

Browse files
Files changed (1) hide show
  1. modules/whisper/whisper_base.py +6 -10
modules/whisper/whisper_base.py CHANGED
@@ -413,10 +413,10 @@ class WhisperBase(ABC):
413
  title_line = ""
414
  if add_timestamp_preview: title_line += "□"
415
  if diarize_speakers: title_line += "□"
416
- title_line += f'Media file:\t{info["input_source_file"]}\n'
417
- title_line += f'Language:\t{info["lang"]} (probability {info["lang_prob"]}%)\n'
418
- if params.is_translate: title_line += f'Translation:\t{info["transcription"]} (Handled by OpenAI Whisper)\n'
419
- if translate_output: title_line += f'Translation:\t{info["translation"]} (Handled by Facebook NLLB)\n'
420
  title_line += "■"
421
 
422
  temp_subtitle = info["subtitle"]
@@ -427,12 +427,8 @@ class WhisperBase(ABC):
427
  total_time += info["time_for_task"]
428
 
429
  time_end = datetime.now()
430
- #total_info += f"\nTotal processing time:\t{self.format_time((time_end-time_start).total_seconds())}"
431
-
432
- temp_file_count_text = "file"
433
- if file_count!=1:
434
- temp_file_count_text += "s"
435
- total_info = f"\nProcessed {file_count} {'file' if file_count == 1 else 'files'} in {self.format_time((time_end-time_start).total_seconds())}"
436
 
437
  result_str = total_result.rstrip("\n")
438
  result_str = self.transform_text_to_list(result_str,"□","■")
 
413
  title_line = ""
414
  if add_timestamp_preview: title_line += "□"
415
  if diarize_speakers: title_line += "□"
416
+ title_line += f'Media file:\t{info["input_source_file"]}'
417
+ title_line += f'\nLanguage:\t{info["lang"]} (probability {info["lang_prob"]}%)'
418
+ if params.is_translate: title_line += f'\nTranslation:\t{info["transcription"]} (Handled by OpenAI Whisper)'
419
+ if translate_output: title_line += f'\nTranslation:\t{info["translation"]} (Handled by Facebook NLLB)'
420
  title_line += "■"
421
 
422
  temp_subtitle = info["subtitle"]
 
427
  total_time += info["time_for_task"]
428
 
429
  time_end = datetime.now()
430
+ #total_info += f"Total processing time:\t{self.format_time((time_end-time_start).total_seconds())}"
431
+ total_info = f"Processed {file_count} {'file' if file_count == 1 else 'files'} in {self.format_time((time_end-time_start).total_seconds())}"
 
 
 
 
432
 
433
  result_str = total_result.rstrip("\n")
434
  result_str = self.transform_text_to_list(result_str,"□","■")