Update app.py

app.py

@@ -22,6 +22,7 @@ import time
 from concurrent.futures import ThreadPoolExecutor, as_completed
 from threading import Lock
 import scipy.io.wavfile
+import spaces
 
 # Logging setup
 logging.basicConfig(level=logging.INFO)
@@ -371,6 +372,7 @@ def download_audio(url, cookie_file=None):
             os.remove(temp_output_path)
             logger.info(f"Temporary file deleted: {temp_output_path}")
 
+@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)):
     if not audio:
         raise ValueError("No audio file provided.")
@@ -433,6 +435,7 @@ def roformer_separator(audio, model_key, seg_size, override_seg_size, overlap, p
             torch.cuda.empty_cache()
             logger.info("GPU memory cleared")
 
+@spaces.GPU
 def auto_ensemble_process(audio, model_keys, seg_size=64, overlap=0.1, out_format="wav", use_tta="False", model_dir="/tmp/audio-separator-models/", output_dir="output", norm_thresh=0.9, amp_thresh=0.9, batch_size=1, ensemble_method="avg_wave", exclude_stems="", weights_str="", progress=gr.Progress(track_tqdm=True)):
     temp_audio_path = None
     max_retries = 2