Spaces:
Running
Running
jhj0517
commited on
Commit
·
b62a259
1
Parent(s):
e4665e3
Initialize output paths
Browse files
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]
|