Update app.py

app.py

@@ -9,6 +9,7 @@ from transformers import MusicgenForConditionalGeneration, MusicgenProcessor, se
 from transformers.generation.streamers import BaseStreamer
 
 import gradio as gr
+import spaces
 
 
 device = "cuda:0" if torch.cuda.is_available() else "cpu"
@@ -143,6 +144,7 @@ sampling_rate = model.audio_encoder.config.sampling_rate
 frame_rate = model.audio_encoder.config.frame_rate
 
 
+@spaces.GPU
 def generate_audio(text_prompt, audio_length_in_s=10.0, play_steps_in_s=2.0):
     inputs = processor(
         text=text_prompt,