Update modules/whisper/whisper_base.py
Browse files
modules/whisper/whisper_base.py
CHANGED
@@ -329,12 +329,12 @@ class WhisperBase(ABC):
|
|
329 |
file_name, file_ext = os.path.splitext(os.path.basename(file))
|
330 |
|
331 |
if preview_format == "text":
|
332 |
-
subtitle = get_txt(transcribed_segments)
|
333 |
-
if preview_format == "text with timestamps":
|
334 |
subtitle = get_plaintext(transcribed_segments)
|
335 |
-
if preview_format == "
|
|
|
|
|
336 |
subtitle = get_csv(transcribed_segments)
|
337 |
-
|
338 |
files_info[file_name] = {"subtitle": subtitle, "time_for_task": time_for_task, "lang": file_language, "lang_prob": file_lang_probs, "input_source_file": (file_name+file_ext), "translation": translation_note, "transcription": transcription_note}
|
339 |
|
340 |
## Add output file as txt
|
|
|
329 |
file_name, file_ext = os.path.splitext(os.path.basename(file))
|
330 |
|
331 |
if preview_format == "text":
|
|
|
|
|
332 |
subtitle = get_plaintext(transcribed_segments)
|
333 |
+
if preview_format == "text with timestamps":
|
334 |
+
subtitle = get_txt(transcribed_segments)
|
335 |
+
if preview_format == "csv":
|
336 |
subtitle = get_csv(transcribed_segments)
|
337 |
+
|
338 |
files_info[file_name] = {"subtitle": subtitle, "time_for_task": time_for_task, "lang": file_language, "lang_prob": file_lang_probs, "input_source_file": (file_name+file_ext), "translation": translation_note, "transcription": transcription_note}
|
339 |
|
340 |
## Add output file as txt
|