Update app.py

app.py

@@ -4,6 +4,7 @@ import soundfile as sf
 from huggingface_hub import login
 from diffusers import StableAudioPipeline
 import gradio as gr
+import spaces
 
 # Load Hugging Face token securely
 HUGGINGFACE_TOKEN = os.getenv("HF_TOKEN")
@@ -28,6 +29,7 @@ pipe = StableAudioPipeline.from_pretrained(
 pipe = pipe.to(device)
 
 # Function to generate audio
+@spaces.GPU
 def generate_audio(prompt, negative_prompt, duration, diffusion_steps, seed):
     generator = torch.Generator(device).manual_seed(seed)
     audio_output = pipe(