enable zerogpu

app.py

@@ -5,6 +5,7 @@ from demucs import pretrained
 from demucs.apply import apply_model
 from pyharp import *
 from audiotools import AudioSignal
+import spaces 
 
 # Available Demucs models
 DEMUX_MODELS = ["mdx_extra_q", "mdx_extra", "htdemucs", "mdx_q"]
@@ -17,7 +18,7 @@ STEM_CHOICES = {
     "Instrumental (No Vocals)": "instrumental"
 }
 
-
+@spaces.GPU
 def separate_stem(audio_file_path: str, model_name: str, stem_choice: str):
     """
     Separates an audio file into the chosen stem using a Demucs model.
@@ -65,7 +66,6 @@ def separate_stem(audio_file_path: str, model_name: str, stem_choice: str):
     stem_signal = AudioSignal(stem.cpu().numpy().astype('float32'), sample_rate=sr)
     return stem_signal
 
-
 def process_fn_stem(audio_file_path: str, demucs_model: str, stem_choice: str):
     """
     PyHARP process function: