TheStinger commited on
Commit
b0592e6
·
verified ·
1 Parent(s): 32910d9

Update app.py

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