Update modules/whisper/whisper_base.py
Browse files
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(
|
|
|
|
|
|
|
|
|
|
|
|
|
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 = ""
|