next-playground commited on
Commit
b26e793
·
verified ·
1 Parent(s): 1ebfaf5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
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, output_dir="/tmp")
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, output_dir="/tmp")
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, output_dir="/tmp")
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]