LAP-DEV commited on
Commit
fdfc2bb
·
verified ·
1 Parent(s): e6c03ad

Update modules/whisper/whisper_base.py

Browse files
Files changed (1) hide show
  1. modules/whisper/whisper_base.py +3 -5
modules/whisper/whisper_base.py CHANGED
@@ -413,17 +413,16 @@ 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"]}'
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"]
423
  temp_subtitle = temp_subtitle.replace("\t","□")
424
  temp_subtitle = temp_subtitle.replace("\n","■")
425
- total_result += title_line
426
- total_result += temp_subtitle
427
 
428
  total_time += info["time_for_task"]
429
 
@@ -431,7 +430,6 @@ class WhisperBase(ABC):
431
  #total_info += f"Total processing time:\t{self.format_time((time_end-time_start).total_seconds())}"
432
  total_info = f"Processed {file_count} {'file' if file_count == 1 else 'files'} in {self.format_time((time_end-time_start).total_seconds())}"
433
 
434
- result_str = total_result.rstrip("\n")
435
  result_str = self.transform_text_to_list(result_str,"□","■")
436
  result_file_path = [info['path'] for info in files_to_download.values()]
437
 
 
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"]rstrip("\n")
423
  temp_subtitle = temp_subtitle.replace("\t","□")
424
  temp_subtitle = temp_subtitle.replace("\n","■")
425
+ total_result += title_line+temp_subtitle
 
426
 
427
  total_time += info["time_for_task"]
428
 
 
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 = self.transform_text_to_list(result_str,"□","■")
434
  result_file_path = [info['path'] for info in files_to_download.values()]
435