import gradio as gr
from diffusers import DiffusionPipeline

model_name = "facebook/musicgen-small"  
pipe = DiffusionPipeline.from_pretrained(model_name)

def generate(prompt):
    audio = pipe(prompt)
    return audio

demo = gr.Interface(fn = generate, inputs="text", outputs=audio.save("output_music.wav"))

demo.launch()