| 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() | |