File size: 347 Bytes
506f9b8 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 | 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()
|