jhj0517 commited on
Commit
3b27598
·
1 Parent(s): e76bbe3

Add lrc extension

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. modules/utils/subtitle_manager.py +1 -0
app.py CHANGED
@@ -53,7 +53,7 @@ class App:
53
  dd_lang = gr.Dropdown(choices=self.whisper_inf.available_langs + [AUTOMATIC_DETECTION],
54
  value=AUTOMATIC_DETECTION if whisper_params["lang"] == AUTOMATIC_DETECTION.unwrap()
55
  else whisper_params["lang"], label=_("Language"))
56
- dd_file_format = gr.Dropdown(choices=["SRT", "WebVTT", "txt"], value="SRT", label=_("File Format"))
57
  with gr.Row():
58
  cb_translate = gr.Checkbox(value=whisper_params["is_translate"], label=_("Translate to English?"),
59
  interactive=True)
 
53
  dd_lang = gr.Dropdown(choices=self.whisper_inf.available_langs + [AUTOMATIC_DETECTION],
54
  value=AUTOMATIC_DETECTION if whisper_params["lang"] == AUTOMATIC_DETECTION.unwrap()
55
  else whisper_params["lang"], label=_("Language"))
56
+ dd_file_format = gr.Dropdown(choices=["SRT", "WebVTT", "txt", "LRC"], value="SRT", label=_("File Format"))
57
  with gr.Row():
58
  cb_translate = gr.Checkbox(value=whisper_params["is_translate"], label=_("Translate to English?"),
59
  interactive=True)
modules/utils/subtitle_manager.py CHANGED
@@ -358,6 +358,7 @@ def get_writer(
358
  "srt": WriteSRT,
359
  "tsv": WriteTSV,
360
  "json": WriteJSON,
 
361
  }
362
 
363
  if output_format == "all":
 
358
  "srt": WriteSRT,
359
  "tsv": WriteTSV,
360
  "json": WriteJSON,
361
+ "lrc": WriteLRC
362
  }
363
 
364
  if output_format == "all":