Update app.py

app.py

@@ -9,6 +9,9 @@ from diffusers import DDPMScheduler
 from pico_model import PicoDiffusion
 from audioldm.variational_autoencoder.autoencoder import AutoencoderKL
 from llm_preprocess import get_event, preprocess_gemini, preprocess_gpt
+
+import spaces
+
 class dotdict(dict):
     """dot.notation access to dictionary attributes"""
     __getattr__ = dict.get
@@ -35,6 +38,8 @@ class InferRunner:
 device = "cuda" if torch.cuda.is_available() else "cpu"
 runner = InferRunner(device)
 event_list = get_event()
+
+@spaces.GPU(duration=240)
 def infer(caption, num_steps=200, guidance_scale=3.0, audio_len=16000*10):
     with torch.no_grad():
         latents = runner.pico_model.demo_inference(caption, runner.scheduler, num_steps=num_steps, guidance_scale=guidance_scale, num_samples_per_prompt=1, disable_progress=True)