Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -7,7 +7,7 @@ from pathlib import Path
|
|
7 |
import threading
|
8 |
import time
|
9 |
|
10 |
-
separator = Separator()
|
11 |
|
12 |
roformer_models = {
|
13 |
'BS-Roformer-Viperx-1297.ckpt': 'model_bs_roformer_ep_317_sdr_12.9755.ckpt',
|
@@ -102,21 +102,21 @@ demucs_models = [
|
|
102 |
def roformer_separator(audio, checkpoint_name):
|
103 |
full_checkpoint_name = roformer_models[checkpoint_name]
|
104 |
separator.load_model(full_checkpoint_name)
|
105 |
-
output_files = separator.separate(audio
|
106 |
stem1 = output_files[0]
|
107 |
stem2 = output_files[1]
|
108 |
return stem1, stem2
|
109 |
|
110 |
def mdx_vr_separator(audio, checkpoint_name):
|
111 |
separator.load_model(checkpoint_name)
|
112 |
-
output_files = separator.separate(audio
|
113 |
stem1 = output_files[0]
|
114 |
stem2 = output_files[1]
|
115 |
return stem1, stem2
|
116 |
|
117 |
def demucs_separator(audio, checkpoint_name):
|
118 |
separator.load_model(checkpoint_name)
|
119 |
-
output_files = separator.separate(audio
|
120 |
stem1 = output_files[0]
|
121 |
stem2 = output_files[1]
|
122 |
stem3 = output_files[2]
|
|
|
7 |
import threading
|
8 |
import time
|
9 |
|
10 |
+
separator = Separator(output_dir="/tmp")
|
11 |
|
12 |
roformer_models = {
|
13 |
'BS-Roformer-Viperx-1297.ckpt': 'model_bs_roformer_ep_317_sdr_12.9755.ckpt',
|
|
|
102 |
def roformer_separator(audio, checkpoint_name):
|
103 |
full_checkpoint_name = roformer_models[checkpoint_name]
|
104 |
separator.load_model(full_checkpoint_name)
|
105 |
+
output_files = separator.separate(audio)
|
106 |
stem1 = output_files[0]
|
107 |
stem2 = output_files[1]
|
108 |
return stem1, stem2
|
109 |
|
110 |
def mdx_vr_separator(audio, checkpoint_name):
|
111 |
separator.load_model(checkpoint_name)
|
112 |
+
output_files = separator.separate(audio)
|
113 |
stem1 = output_files[0]
|
114 |
stem2 = output_files[1]
|
115 |
return stem1, stem2
|
116 |
|
117 |
def demucs_separator(audio, checkpoint_name):
|
118 |
separator.load_model(checkpoint_name)
|
119 |
+
output_files = separator.separate(audio)
|
120 |
stem1 = output_files[0]
|
121 |
stem2 = output_files[1]
|
122 |
stem3 = output_files[2]
|