Update app.py

app.py

@@ -1,6 +1,6 @@
 import gradio as gr
 from transformers import pipeline
-
+from IPython.display import Audio
 from transformers import AutoProcessor, MusicgenForConditionalGeneration
 
 processor = AutoProcessor.from_pretrained("facebook/musicgen-small")
@@ -10,7 +10,8 @@ model = MusicgenForConditionalGeneration.from_pretrained("facebook/musicgen-smal
 
 def generate(description):
     audio = processor(description, padding=True, return_tensors="pt")
-    return audio['audio']
+    audio_values = model.generate(**audio, max_new_tokens=256)
+    return Audio(audio_values[0].numpy(), rate=sampling_rate)
 
 demo = gr.Interface(
     fn=generate,