Update app.py

app.py

@@ -0,0 +1,18 @@
+import gradio as gr
+from diffusers import DDPMPipeline
+
+# Load model
+pipe = DDPMPipeline.from_pretrained("google/ddpm-mnist")
+
+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()