Update app.py

app.py

@@ -453,6 +453,7 @@ def download_audio(url, cookies_file=None, out_dir="ytdl"):
         logger.error(f"Download failed: {e}")
         raise RuntimeError(f"Download failed: {e}")
 
+@spaces.GPU
 def roformer_separator(audio, model_key, seg_size, override_seg_size, overlap, pitch_shift, model_dir, output_dir, out_format, norm_thresh, amp_thresh, batch_size, exclude_stems="", progress=gr.Progress(track_tqdm=True)):
     """Separate audio into stems using a Roformer model."""
     if not audio:
@@ -499,6 +500,7 @@ def roformer_separator(audio, model_key, seg_size, override_seg_size, overlap, p
         logger.error(f"Separation failed: {e}")
         raise RuntimeError(f"Separation failed: {e}")
 
+@spaces.GPU
 def auto_ensemble_process(audio, model_keys, seg_size, overlap, out_format, use_tta, model_dir, output_dir, norm_thresh, amp_thresh, batch_size, ensemble_method, exclude_stems="", weights_str="", progress=gr.Progress()):
     """Perform ensemble processing on audio using multiple Roformer models."""
     if not audio or not model_keys: