LAP-DEV commited on
Commit
ebfa7b3
·
verified ·
1 Parent(s): a3eaf18

Update modules/whisper/whisper_base.py

Browse files
Files changed (1) hide show
  1. modules/whisper/whisper_base.py +4 -4
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 == "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
 
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