import gradio as gr
from diffusers import DDPMPipeline

# Load model
pipe = DDPMPipeline.from_pretrained("google/ddpm-cifar10-32")
def generate():
    image = pipe().images[0]
    return image

# Create UI
gr.Interface(
    fn=generate,
    inputs=[],
    outputs="image",
    title="MNIST Diffusion Generator",
    description="Click button to generate handwritten digit using diffusion model"
).launch()