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