jhj0517 commited on
Commit
598f1d1
·
1 Parent(s): 5306f0a

Embed bgm seperator inside whisper base

Browse files
Files changed (1) hide show
  1. app.py +3 -7
app.py CHANGED
@@ -36,10 +36,6 @@ class App:
36
  self.deepl_api = DeepLAPI(
37
  output_dir=os.path.join(self.args.output_dir, "translations")
38
  )
39
- self.music_separator = MusicSeparator(
40
- model_dir=self.args.uvr_model_dir,
41
- output_dir=os.path.join(self.args.output_dir, "UVR")
42
- )
43
  self.default_params = load_yaml(DEFAULT_PARAMETERS_CONFIG_PATH)
44
  print(f"Use \"{self.args.whisper_type}\" implementation")
45
  print(f"Device \"{self.whisper_inf.device}\" is detected")
@@ -154,10 +150,10 @@ class App:
154
  with gr.Accordion("BGM Separation", open=False):
155
  cb_bgm_separation = gr.Checkbox(label="Enable BGM separation", value=uvr_params["is_separate_bgm"],
156
  interactive=True)
157
- dd_uvr_device = gr.Dropdown(label="Device", value=self.music_separator.device,
158
- choices=self.music_separator.available_devices)
159
  dd_uvr_model_size = gr.Dropdown(label="Model", value=uvr_params["model_size"],
160
- choices=self.music_separator.available_models)
161
  nb_uvr_segment_size = gr.Number(label="Segment Size", value=uvr_params["segment_size"], precision=0)
162
  cb_uvr_save_file = gr.Checkbox(label="Save Separated Files to Output Directory",
163
  value=uvr_params["save_file"], interactive=True)
 
36
  self.deepl_api = DeepLAPI(
37
  output_dir=os.path.join(self.args.output_dir, "translations")
38
  )
 
 
 
 
39
  self.default_params = load_yaml(DEFAULT_PARAMETERS_CONFIG_PATH)
40
  print(f"Use \"{self.args.whisper_type}\" implementation")
41
  print(f"Device \"{self.whisper_inf.device}\" is detected")
 
150
  with gr.Accordion("BGM Separation", open=False):
151
  cb_bgm_separation = gr.Checkbox(label="Enable BGM separation", value=uvr_params["is_separate_bgm"],
152
  interactive=True)
153
+ dd_uvr_device = gr.Dropdown(label="Device", value=self.whisper_inf.music_separator.device,
154
+ choices=self.whisper_inf.music_separator.available_devices)
155
  dd_uvr_model_size = gr.Dropdown(label="Model", value=uvr_params["model_size"],
156
+ choices=self.whisper_inf.music_separator.available_models)
157
  nb_uvr_segment_size = gr.Number(label="Segment Size", value=uvr_params["segment_size"], precision=0)
158
  cb_uvr_save_file = gr.Checkbox(label="Save Separated Files to Output Directory",
159
  value=uvr_params["save_file"], interactive=True)