File size: 453 Bytes
ada9672 dcee4d4 ada9672 dcee4d4 ada9672 dcee4d4 ada9672 dcee4d4 ada9672 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 |
from diffusers import DiffusionPipeline
import gradio as gr
pipe = DiffusionPipeline.from_pretrained("riffusion/riffusion-model-v1")
pipe.to("cpu") # Safe for any system
def generate_music(prompt):
output = pipe(prompt)
audio = output["audio"]
return audio
gr.Interface(
fn=generate_music,
inputs=gr.Textbox(label="Describe your music"),
outputs=gr.Audio(label="AI Song"),
title="LarynxLab - Riffusion Edition"
).launch() |