jhj0517 commited on
Commit
b253e57
·
1 Parent(s): 7c12536

do not use dataclass

Browse files
Files changed (1) hide show
  1. app.py +29 -26
app.py CHANGED
@@ -132,34 +132,37 @@ class App:
132
  files_subtitles = gr.Files(label="Downloadable output file", scale=3, interactive=False)
133
 
134
  params = [input_file, dd_file_format, cb_timestamp]
135
- whisper_params = WhisperParameters(model_size=dd_model,
136
- lang=dd_lang,
137
- is_translate=cb_translate,
138
- beam_size=nb_beam_size,
139
- log_prob_threshold=nb_log_prob_threshold,
140
- no_speech_threshold=nb_no_speech_threshold,
141
- compute_type=dd_compute_type,
142
- best_of=nb_best_of,
143
- patience=nb_patience,
144
- condition_on_previous_text=cb_condition_on_previous_text,
145
- initial_prompt=tb_initial_prompt,
146
- temperature=sd_temperature,
147
- compression_ratio_threshold=nb_compression_ratio_threshold,
148
- vad_filter=cb_vad_filter,
149
- threshold=sd_threshold,
150
- min_speech_duration_ms=nb_min_speech_duration_ms,
151
- max_speech_duration_s=nb_max_speech_duration_s,
152
- min_silence_duration_ms=nb_min_silence_duration_ms,
153
- window_size_sample=nb_window_size_sample,
154
- speech_pad_ms=nb_speech_pad_ms,
155
- chunk_length_s=nb_chunk_length_s,
156
- batch_size=nb_batch_size,
157
- is_diarize=cb_diarize,
158
- hf_token=tb_hf_token,
159
- diarization_device=dd_diarization_device)
 
 
 
160
 
161
  btn_run.click(fn=self.whisper_inf.transcribe_file,
162
- inputs=params + whisper_params.as_list(),
163
  outputs=[tb_indicator, files_subtitles])
164
  dd_model.change(fn=self.on_change_models, inputs=[dd_model], outputs=[cb_translate])
165
 
 
132
  files_subtitles = gr.Files(label="Downloadable output file", scale=3, interactive=False)
133
 
134
  params = [input_file, dd_file_format, cb_timestamp]
135
+
136
+ whisper_params = [
137
+ dd_model,
138
+ dd_lang,
139
+ cb_translate,
140
+ nb_beam_size,
141
+ nb_log_prob_threshold,
142
+ nb_no_speech_threshold,
143
+ dd_compute_type,
144
+ nb_best_of,
145
+ nb_patience,
146
+ cb_condition_on_previous_text,
147
+ tb_initial_prompt,
148
+ sd_temperature,
149
+ nb_compression_ratio_threshold,
150
+ cb_vad_filter,
151
+ sd_threshold,
152
+ nb_min_speech_duration_ms,
153
+ nb_max_speech_duration_s,
154
+ nb_min_silence_duration_ms,
155
+ nb_window_size_sample,
156
+ nb_speech_pad_ms,
157
+ nb_chunk_length_s,
158
+ nb_batch_size,
159
+ cb_diarize,
160
+ tb_hf_token,
161
+ dd_diarization_device
162
+ ]
163
 
164
  btn_run.click(fn=self.whisper_inf.transcribe_file,
165
+ inputs=params + whisper_params,
166
  outputs=[tb_indicator, files_subtitles])
167
  dd_model.change(fn=self.on_change_models, inputs=[dd_model], outputs=[cb_translate])
168