jhj0517 commited on
Commit
4df5628
·
1 Parent(s): 806824b

Rename function and variable

Browse files
Files changed (1) hide show
  1. app.py +7 -7
app.py CHANGED
@@ -44,7 +44,7 @@ class App:
44
  print(f"Use \"{self.args.whisper_type}\" implementation\n"
45
  f"Device \"{self.whisper_inf.device}\" is detected")
46
 
47
- def create_whisper_parameters(self):
48
  whisper_params = self.default_params["whisper"]
49
  vad_params = self.default_params["vad"]
50
  diarization_params = self.default_params["diarization"]
@@ -108,7 +108,7 @@ class App:
108
  visible=self.args.colab,
109
  value="")
110
 
111
- whisper_params, dd_file_format, cb_timestamp = self.create_whisper_parameters()
112
 
113
  with gr.Row():
114
  btn_run = gr.Button(_("GENERATE SUBTITLE FILE"), variant="primary")
@@ -119,7 +119,7 @@ class App:
119
 
120
  params = [input_file, tb_input_folder, dd_file_format, cb_timestamp]
121
  btn_run.click(fn=self.whisper_inf.transcribe_file,
122
- inputs=params + whisper_params,
123
  outputs=[tb_indicator, files_subtitles])
124
  btn_openfolder.click(fn=lambda: self.open_folder("outputs"), inputs=None, outputs=None)
125
 
@@ -133,7 +133,7 @@ class App:
133
  tb_title = gr.Label(label=_("Youtube Title"))
134
  tb_description = gr.Textbox(label=_("Youtube Description"), max_lines=15)
135
 
136
- whisper_params, dd_file_format, cb_timestamp = self.create_whisper_parameters()
137
 
138
  with gr.Row():
139
  btn_run = gr.Button(_("GENERATE SUBTITLE FILE"), variant="primary")
@@ -145,7 +145,7 @@ class App:
145
  params = [tb_youtubelink, dd_file_format, cb_timestamp]
146
 
147
  btn_run.click(fn=self.whisper_inf.transcribe_youtube,
148
- inputs=params + whisper_params,
149
  outputs=[tb_indicator, files_subtitles])
150
  tb_youtubelink.change(get_ytmetas, inputs=[tb_youtubelink],
151
  outputs=[img_thumbnail, tb_title, tb_description])
@@ -155,7 +155,7 @@ class App:
155
  with gr.Row():
156
  mic_input = gr.Microphone(label=_("Record with Mic"), type="filepath", interactive=True)
157
 
158
- whisper_params, dd_file_format, cb_timestamp = self.create_whisper_parameters()
159
 
160
  with gr.Row():
161
  btn_run = gr.Button(_("GENERATE SUBTITLE FILE"), variant="primary")
@@ -167,7 +167,7 @@ class App:
167
  params = [mic_input, dd_file_format, cb_timestamp]
168
 
169
  btn_run.click(fn=self.whisper_inf.transcribe_mic,
170
- inputs=params + whisper_params,
171
  outputs=[tb_indicator, files_subtitles])
172
  btn_openfolder.click(fn=lambda: self.open_folder("outputs"), inputs=None, outputs=None)
173
 
 
44
  print(f"Use \"{self.args.whisper_type}\" implementation\n"
45
  f"Device \"{self.whisper_inf.device}\" is detected")
46
 
47
+ def create_pipeline_inputs(self):
48
  whisper_params = self.default_params["whisper"]
49
  vad_params = self.default_params["vad"]
50
  diarization_params = self.default_params["diarization"]
 
108
  visible=self.args.colab,
109
  value="")
110
 
111
+ pipeline_params, dd_file_format, cb_timestamp = self.create_pipeline_inputs()
112
 
113
  with gr.Row():
114
  btn_run = gr.Button(_("GENERATE SUBTITLE FILE"), variant="primary")
 
119
 
120
  params = [input_file, tb_input_folder, dd_file_format, cb_timestamp]
121
  btn_run.click(fn=self.whisper_inf.transcribe_file,
122
+ inputs=params + pipeline_params,
123
  outputs=[tb_indicator, files_subtitles])
124
  btn_openfolder.click(fn=lambda: self.open_folder("outputs"), inputs=None, outputs=None)
125
 
 
133
  tb_title = gr.Label(label=_("Youtube Title"))
134
  tb_description = gr.Textbox(label=_("Youtube Description"), max_lines=15)
135
 
136
+ pipeline_params, dd_file_format, cb_timestamp = self.create_pipeline_inputs()
137
 
138
  with gr.Row():
139
  btn_run = gr.Button(_("GENERATE SUBTITLE FILE"), variant="primary")
 
145
  params = [tb_youtubelink, dd_file_format, cb_timestamp]
146
 
147
  btn_run.click(fn=self.whisper_inf.transcribe_youtube,
148
+ inputs=params + pipeline_params,
149
  outputs=[tb_indicator, files_subtitles])
150
  tb_youtubelink.change(get_ytmetas, inputs=[tb_youtubelink],
151
  outputs=[img_thumbnail, tb_title, tb_description])
 
155
  with gr.Row():
156
  mic_input = gr.Microphone(label=_("Record with Mic"), type="filepath", interactive=True)
157
 
158
+ pipeline_params, dd_file_format, cb_timestamp = self.create_pipeline_inputs()
159
 
160
  with gr.Row():
161
  btn_run = gr.Button(_("GENERATE SUBTITLE FILE"), variant="primary")
 
167
  params = [mic_input, dd_file_format, cb_timestamp]
168
 
169
  btn_run.click(fn=self.whisper_inf.transcribe_mic,
170
+ inputs=params + pipeline_params,
171
  outputs=[tb_indicator, files_subtitles])
172
  btn_openfolder.click(fn=lambda: self.open_folder("outputs"), inputs=None, outputs=None)
173