Update app.py
Browse files
app.py
CHANGED
|
@@ -257,6 +257,9 @@ class App:
|
|
| 257 |
|
| 258 |
#dd_model.change(fn=self.on_change_models, inputs=[dd_model], outputs=[cb_translate])
|
| 259 |
|
|
|
|
|
|
|
|
|
|
| 260 |
return (
|
| 261 |
WhisperParameters(
|
| 262 |
model_size=dd_model, lang=dd_lang, is_translate=cb_translate, beam_size=nb_beam_size,
|
|
@@ -278,7 +281,8 @@ class App:
|
|
| 278 |
language_detection_segments=nb_language_detection_segments,
|
| 279 |
prompt_reset_on_temperature=sld_prompt_reset_on_temperature, is_bgm_separate=cb_bgm_separation,
|
| 280 |
uvr_device=dd_uvr_device, uvr_model_size=dd_uvr_model_size, uvr_segment_size=nb_uvr_segment_size,
|
| 281 |
-
uvr_save_file=cb_uvr_save_file, uvr_enable_offload=cb_uvr_enable_offload,
|
|
|
|
| 282 |
),
|
| 283 |
dd_file_format,
|
| 284 |
cb_timestamp,
|
|
|
|
| 257 |
|
| 258 |
#dd_model.change(fn=self.on_change_models, inputs=[dd_model], outputs=[cb_translate])
|
| 259 |
|
| 260 |
+
bool_whisper_enable_offload = whisper_params["enable_offload"]
|
| 261 |
+
bool_diarization_enable_offload = diarization_params["enable_offload"]
|
| 262 |
+
|
| 263 |
return (
|
| 264 |
WhisperParameters(
|
| 265 |
model_size=dd_model, lang=dd_lang, is_translate=cb_translate, beam_size=nb_beam_size,
|
|
|
|
| 281 |
language_detection_segments=nb_language_detection_segments,
|
| 282 |
prompt_reset_on_temperature=sld_prompt_reset_on_temperature, is_bgm_separate=cb_bgm_separation,
|
| 283 |
uvr_device=dd_uvr_device, uvr_model_size=dd_uvr_model_size, uvr_segment_size=nb_uvr_segment_size,
|
| 284 |
+
uvr_save_file=cb_uvr_save_file, uvr_enable_offload=cb_uvr_enable_offload,
|
| 285 |
+
whisper_enable_offload=bool_whisper_enable_offload, diarization_enable_offload=bool_diarization_enable_offload
|
| 286 |
),
|
| 287 |
dd_file_format,
|
| 288 |
cb_timestamp,
|