Update app.py

app.py

@@ -2,7 +2,9 @@ import gradio as gr
 import torchaudio
 from audiocraft.models import MusicGen
 from audiocraft.data.audio import audio_write
+import spaces
 
+@spaces.GPU(duration=120)
 def generate_music(description, melody_audio):
     model = MusicGen.get_pretrained('nateraw/musicgen-songstarter-v0.2')
     model.set_generation_params(duration=8)  # generate 8 seconds.