Update app.py

app.py

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