Spaces:
Runtime error
Runtime error
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]
|