Update app.py

app.py

@@ -41,6 +41,7 @@ def get_cached_model(model_name: str):
     return LOADED_MODELS[model_name]
 
 # Separation Logic
+@spaces.GPU
 def separate_all_stems(audio_file_path: str, model_name: str):
     model = get_cached_model(model_name)
 
@@ -94,7 +95,6 @@ def separate_all_stems(audio_file_path: str, model_name: str):
     return [drums, bass, vocals, other]
 
 # Process Function
-@spaces.GPU
 def process_fn(audio_file_path, model_name):
     try:
         if model_name not in DEMUX_MODELS: