LAP-DEV commited on
Commit
b171e66
·
verified ·
1 Parent(s): 0086005

Update modules/whisper/whisper_base.py

Browse files
Files changed (1) hide show
  1. modules/whisper/whisper_base.py +7 -1
modules/whisper/whisper_base.py CHANGED
@@ -362,7 +362,13 @@ class WhisperBase(ABC):
362
  ## Add output file as txt, srt and/or csv
363
  for output_format in file_format:
364
  file_name, file_ext = os.path.splitext(os.path.basename(file))
365
- subtitle, file_path = self.generate_and_write_file(file_name,transcribed_segments,add_timestamp,output_format.lower(),self.output_dir)
 
 
 
 
 
 
366
  files_to_download[file_name+"_"+output_format.lower()] = {"path": file_path}
367
 
368
  total_result = ""
 
362
  ## Add output file as txt, srt and/or csv
363
  for output_format in file_format:
364
  file_name, file_ext = os.path.splitext(os.path.basename(file))
365
+ subtitle, file_path = self.generate_and_write_file(
366
+ file_name=file_name,
367
+ transcribed_segments=transcribed_segments,
368
+ add_timestamp=add_timestamp,
369
+ file_format=format.lower(),
370
+ output_dir=self.output_dir
371
+ )
372
  files_to_download[file_name+"_"+output_format.lower()] = {"path": file_path}
373
 
374
  total_result = ""