fixed bugs

app.py

@@ -8,6 +8,7 @@ from models import AudioDiffusion, DDPMScheduler
 from audioldm.audio.stft import TacotronSTFT
 from audioldm.variational_autoencoder import AutoencoderKL
 from gradio import Markdown
+import spaces
 
 class Tango:
     def __init__(self, name="declare-lab/tango2", device="cuda:0"):
@@ -71,7 +72,8 @@ if torch.cuda.is_available():
     tango = Tango()
 else:
     tango = Tango(device="cpu")
-
+    
+@spaces.GPU(duration=240)
 def gradio_generate(prompt, steps, guidance):
     output_wave = tango.generate(prompt, steps, guidance)
     # output_filename = f"{prompt.replace(' ', '_')}_{steps}_{guidance}"[:250] + ".wav"