jhj0517 commited on
Commit
b62a259
·
1 Parent(s): e4665e3

Initialize output paths

Browse files
Files changed (1) hide show
  1. modules/uvr/music_separator.py +4 -0
modules/uvr/music_separator.py CHANGED
@@ -22,6 +22,10 @@ class MusicSeparator:
22
  self.available_devices = ["cpu", "cuda"]
23
  self.model_dir = model_dir
24
  self.output_dir = output_dir
 
 
 
 
25
  self.audio_info = None
26
  self.available_models = ["UVR-MDX-NET-Inst_HQ_4", "UVR-MDX-NET-Inst_3"]
27
  self.default_model = self.available_models[0]
 
22
  self.available_devices = ["cpu", "cuda"]
23
  self.model_dir = model_dir
24
  self.output_dir = output_dir
25
+ instrumental_output_dir = os.path.join(self.output_dir, "instrumental")
26
+ vocals_output_dir = os.path.join(self.output_dir, "vocals")
27
+ os.makedirs(instrumental_output_dir, exist_ok=True)
28
+ os.makedirs(vocals_output_dir, exist_ok=True)
29
  self.audio_info = None
30
  self.available_models = ["UVR-MDX-NET-Inst_HQ_4", "UVR-MDX-NET-Inst_3"]
31
  self.default_model = self.available_models[0]