import gradio as gr
from audiocraft.models import MusicGen

model = MusicGen.get_pretrained('small')

def generate_music(prompt):
    wav = model.generate(prompt, progress=True)
    filename = "music.wav"
    model.save_wav(wav, filename)
    return filename

iface = gr.Interface(fn=generate_music, inputs="text", outputs="audio")
iface.launch()