Update app.py

app.py

@@ -379,7 +379,7 @@ def download_audio(url, cookie_file=None):
             os.remove(temp_output_path)
             logger.info(f"Temporary file deleted: {temp_output_path}")
 
-@spaces.GPU
+@spaces.GPU(duration=60)
 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 or video file provided.")
@@ -481,7 +481,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
+@spaces.GPU(duration=60)
 def auto_ensemble_process(audio, model_keys, state, 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
     extracted_audio_path = None