TheStinger commited on
Commit
eeb7234
·
verified ·
1 Parent(s): d9ed2fd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -93,6 +93,7 @@ demucs_models = [
93
  'hdemucs_mmi.yaml',
94
  ]
95
 
 
96
  def roformer_separator(audio, checkpoint_name):
97
  full_checkpoint_name = roformer_models[checkpoint_name]
98
  separator.load_model(full_checkpoint_name)
@@ -101,6 +102,7 @@ def roformer_separator(audio, checkpoint_name):
101
  stem2 = output_files[1]
102
  return stem1, stem2
103
 
 
104
  def mdx_vr_separator(audio, checkpoint_name):
105
  separator.load_model(checkpoint_name)
106
  output_files = separator.separate(audio)
@@ -108,6 +110,7 @@ def mdx_vr_separator(audio, checkpoint_name):
108
  stem2 = output_files[1]
109
  return stem1, stem2
110
 
 
111
  def demucs_separator(audio, checkpoint_name):
112
  separator.load_model(checkpoint_name)
113
  output_files = separator.separate(audio)
 
93
  'hdemucs_mmi.yaml',
94
  ]
95
 
96
+ @spaces.zerogpu
97
  def roformer_separator(audio, checkpoint_name):
98
  full_checkpoint_name = roformer_models[checkpoint_name]
99
  separator.load_model(full_checkpoint_name)
 
102
  stem2 = output_files[1]
103
  return stem1, stem2
104
 
105
+ @spaces.zerogpu
106
  def mdx_vr_separator(audio, checkpoint_name):
107
  separator.load_model(checkpoint_name)
108
  output_files = separator.separate(audio)
 
110
  stem2 = output_files[1]
111
  return stem1, stem2
112
 
113
+ @spaces.zerogpu
114
  def demucs_separator(audio, checkpoint_name):
115
  separator.load_model(checkpoint_name)
116
  output_files = separator.separate(audio)