jhj0517 commited on
Commit
b2f7849
Β·
1 Parent(s): 94c1a40

add "txt" format

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -50,7 +50,7 @@ class App:
50
  label="Model")
51
  dd_lang = gr.Dropdown(choices=["Automatic Detection"] + self.whisper_inf.available_langs,
52
  value="Automatic Detection", label="Language")
53
- dd_subformat = gr.Dropdown(["SRT", "WebVTT"], value="SRT", label="Subtitle Format")
54
  with gr.Row():
55
  cb_translate = gr.Checkbox(value=False, label="Translate to English?", interactive=True)
56
  with gr.Row():
@@ -66,7 +66,7 @@ class App:
66
  tb_indicator = gr.Textbox(label="Output", scale=8)
67
  btn_openfolder = gr.Button('πŸ“‚', scale=2)
68
 
69
- params = [input_file, dd_model, dd_lang, dd_subformat, cb_translate, cb_timestamp]
70
  advanced_params = [nb_beam_size, nb_log_prob_threshold, nb_no_speech_threshold, dd_compute_type]
71
  btn_run.click(fn=self.whisper_inf.transcribe_file,
72
  inputs=params + advanced_params,
@@ -88,7 +88,7 @@ class App:
88
  label="Model")
89
  dd_lang = gr.Dropdown(choices=["Automatic Detection"] + self.whisper_inf.available_langs,
90
  value="Automatic Detection", label="Language")
91
- dd_subformat = gr.Dropdown(choices=["SRT", "WebVTT"], value="SRT", label="Subtitle Format")
92
  with gr.Row():
93
  cb_translate = gr.Checkbox(value=False, label="Translate to English?", interactive=True)
94
  with gr.Row():
@@ -105,7 +105,7 @@ class App:
105
  tb_indicator = gr.Textbox(label="Output", scale=8)
106
  btn_openfolder = gr.Button('πŸ“‚', scale=2)
107
 
108
- params = [tb_youtubelink, dd_model, dd_lang, dd_subformat, cb_translate, cb_timestamp]
109
  advanced_params = [nb_beam_size, nb_log_prob_threshold, nb_no_speech_threshold, dd_compute_type]
110
  btn_run.click(fn=self.whisper_inf.transcribe_youtube,
111
  inputs=params + advanced_params,
@@ -123,7 +123,7 @@ class App:
123
  label="Model")
124
  dd_lang = gr.Dropdown(choices=["Automatic Detection"] + self.whisper_inf.available_langs,
125
  value="Automatic Detection", label="Language")
126
- dd_subformat = gr.Dropdown(["SRT", "WebVTT"], value="SRT", label="Subtitle Format")
127
  with gr.Row():
128
  cb_translate = gr.Checkbox(value=False, label="Translate to English?", interactive=True)
129
  with gr.Accordion("Advanced_Parameters", open=False):
@@ -137,7 +137,7 @@ class App:
137
  tb_indicator = gr.Textbox(label="Output", scale=8)
138
  btn_openfolder = gr.Button('πŸ“‚', scale=2)
139
 
140
- params = [mic_input, dd_model, dd_lang, dd_subformat, cb_translate]
141
  advanced_params = [nb_beam_size, nb_log_prob_threshold, nb_no_speech_threshold, dd_compute_type]
142
  btn_run.click(fn=self.whisper_inf.transcribe_mic,
143
  inputs=params + advanced_params,
 
50
  label="Model")
51
  dd_lang = gr.Dropdown(choices=["Automatic Detection"] + self.whisper_inf.available_langs,
52
  value="Automatic Detection", label="Language")
53
+ dd_file_format = gr.Dropdown(["SRT", "WebVTT", "txt"], value="SRT", label="File Format")
54
  with gr.Row():
55
  cb_translate = gr.Checkbox(value=False, label="Translate to English?", interactive=True)
56
  with gr.Row():
 
66
  tb_indicator = gr.Textbox(label="Output", scale=8)
67
  btn_openfolder = gr.Button('πŸ“‚', scale=2)
68
 
69
+ params = [input_file, dd_model, dd_lang, dd_file_format, cb_translate, cb_timestamp]
70
  advanced_params = [nb_beam_size, nb_log_prob_threshold, nb_no_speech_threshold, dd_compute_type]
71
  btn_run.click(fn=self.whisper_inf.transcribe_file,
72
  inputs=params + advanced_params,
 
88
  label="Model")
89
  dd_lang = gr.Dropdown(choices=["Automatic Detection"] + self.whisper_inf.available_langs,
90
  value="Automatic Detection", label="Language")
91
+ dd_file_format = gr.Dropdown(choices=["SRT", "WebVTT", "txt"], value="SRT", label="File Format")
92
  with gr.Row():
93
  cb_translate = gr.Checkbox(value=False, label="Translate to English?", interactive=True)
94
  with gr.Row():
 
105
  tb_indicator = gr.Textbox(label="Output", scale=8)
106
  btn_openfolder = gr.Button('πŸ“‚', scale=2)
107
 
108
+ params = [tb_youtubelink, dd_model, dd_lang, dd_file_format, cb_translate, cb_timestamp]
109
  advanced_params = [nb_beam_size, nb_log_prob_threshold, nb_no_speech_threshold, dd_compute_type]
110
  btn_run.click(fn=self.whisper_inf.transcribe_youtube,
111
  inputs=params + advanced_params,
 
123
  label="Model")
124
  dd_lang = gr.Dropdown(choices=["Automatic Detection"] + self.whisper_inf.available_langs,
125
  value="Automatic Detection", label="Language")
126
+ dd_file_format = gr.Dropdown(["SRT", "WebVTT", "txt"], value="SRT", label="File Format")
127
  with gr.Row():
128
  cb_translate = gr.Checkbox(value=False, label="Translate to English?", interactive=True)
129
  with gr.Accordion("Advanced_Parameters", open=False):
 
137
  tb_indicator = gr.Textbox(label="Output", scale=8)
138
  btn_openfolder = gr.Button('πŸ“‚', scale=2)
139
 
140
+ params = [mic_input, dd_model, dd_lang, dd_file_format, cb_translate]
141
  advanced_params = [nb_beam_size, nb_log_prob_threshold, nb_no_speech_threshold, dd_compute_type]
142
  btn_run.click(fn=self.whisper_inf.transcribe_mic,
143
  inputs=params + advanced_params,