jhj0517 commited on
Commit
0da25b6
·
1 Parent(s): e9e1347

Pass device as param

Browse files
Files changed (1) hide show
  1. app.py +13 -5
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="faster-whisper",
316
- choices=["whisper", "faster-whisper", "insanely-fast-whisper"],
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')