Update app.py

app.py

@@ -1,5 +1,14 @@
+import streamlit as st
 from diffusers import DDPMPipeline
+import PIL
 
-pipeline = DDPMPipeline.from_pretrained('dvoils/sd-class-butterflies-32')
-image = pipeline().images[0]
-image
+def generate_image():
+    pipeline = DDPMPipeline.from_pretrained('dvoils/sd-class-butterflies-32')
+    image = pipeline().images[0]
+    return image
+
+st.title('DDPMPipeline Image Generator')
+
+if st.button('Generate Image'):
+    image = generate_image()
+    st.image(image, caption='Generated Image', use_column_width=True)