Update app.py

app.py

@@ -1,10 +1,12 @@
 import gradio as gr
+import spaces
 import torchaudio
 from audiocraft.models import AudioGen
 from audiocraft.data.audio import audio_write
 
 model = AudioGen.get_pretrained('facebook/audiogen-medium')
 
+@spaces.GPU(enable_queue=True)
 def infer(prompt):
  
  model.set_generation_params(duration=5) # generate 5 seconds.