Spaces:
Running
Running
jhj0517
commited on
Commit
·
0da25b6
1
Parent(s):
e9e1347
Pass device as param
Browse files
app.py
CHANGED
@@ -66,16 +66,24 @@ class App:
|
|
66 |
interactive=True)
|
67 |
|
68 |
with gr.Accordion(_("Advanced Parameters"), open=False):
|
69 |
-
whisper_inputs = WhisperParams.to_gradio_inputs(defaults=whisper_params, only_advanced=True
|
|
|
|
|
|
|
70 |
|
71 |
with gr.Accordion(_("Background Music Remover Filter"), open=False):
|
72 |
-
uvr_inputs = BGMSeparationParams.to_gradio_input(defaults=uvr_params
|
|
|
|
|
|
|
73 |
|
74 |
with gr.Accordion(_("Voice Detection Filter"), open=False):
|
75 |
vad_inputs = VadParams.to_gradio_inputs(defaults=vad_params)
|
76 |
|
77 |
with gr.Accordion(_("Diarization"), open=False):
|
78 |
-
diarization_inputs = DiarizationParams.to_gradio_inputs(defaults=diarization_params
|
|
|
|
|
79 |
|
80 |
dd_model.change(fn=self.on_change_models, inputs=[dd_model], outputs=[cb_translate])
|
81 |
|
@@ -312,8 +320,8 @@ class App:
|
|
312 |
|
313 |
|
314 |
parser = argparse.ArgumentParser()
|
315 |
-
parser.add_argument('--whisper_type', type=str, default=
|
316 |
-
choices=[
|
317 |
help='A type of the whisper implementation (Github repo name)')
|
318 |
parser.add_argument('--share', type=str2bool, default=False, nargs='?', const=True, help='Gradio share value')
|
319 |
parser.add_argument('--server_name', type=str, default=None, help='Gradio server host')
|
|
|
66 |
interactive=True)
|
67 |
|
68 |
with gr.Accordion(_("Advanced Parameters"), open=False):
|
69 |
+
whisper_inputs = WhisperParams.to_gradio_inputs(defaults=whisper_params, only_advanced=True,
|
70 |
+
whisper_type=self.args.whisper_type,
|
71 |
+
available_compute_types=self.whisper_inf.available_compute_types,
|
72 |
+
compute_type=self.whisper_inf.current_compute_type)
|
73 |
|
74 |
with gr.Accordion(_("Background Music Remover Filter"), open=False):
|
75 |
+
uvr_inputs = BGMSeparationParams.to_gradio_input(defaults=uvr_params,
|
76 |
+
available_models=self.whisper_inf.music_separator.available_models,
|
77 |
+
available_devices=self.whisper_inf.music_separator.available_devices,
|
78 |
+
device=self.whisper_inf.music_separator.device)
|
79 |
|
80 |
with gr.Accordion(_("Voice Detection Filter"), open=False):
|
81 |
vad_inputs = VadParams.to_gradio_inputs(defaults=vad_params)
|
82 |
|
83 |
with gr.Accordion(_("Diarization"), open=False):
|
84 |
+
diarization_inputs = DiarizationParams.to_gradio_inputs(defaults=diarization_params,
|
85 |
+
available_devices=self.whisper_inf.diarizer.available_device,
|
86 |
+
device=self.whisper_inf.diarizer.device)
|
87 |
|
88 |
dd_model.change(fn=self.on_change_models, inputs=[dd_model], outputs=[cb_translate])
|
89 |
|
|
|
320 |
|
321 |
|
322 |
parser = argparse.ArgumentParser()
|
323 |
+
parser.add_argument('--whisper_type', type=str, default=WhisperImpl.FASTER_WHISPER,
|
324 |
+
choices=[item.value for item in WhisperImpl],
|
325 |
help='A type of the whisper implementation (Github repo name)')
|
326 |
parser.add_argument('--share', type=str2bool, default=False, nargs='?', const=True, help='Gradio share value')
|
327 |
parser.add_argument('--server_name', type=str, default=None, help='Gradio server host')
|