Update app.py

app.py

@@ -1,6 +1,7 @@
 import streamlit as st
 from diffusers import DDPMPipeline
 import PIL
+import time
 
 def generate_image():
     pipeline = DDPMPipeline.from_pretrained('dvoils/sd-class-butterflies-32')
@@ -10,5 +11,14 @@ def generate_image():
 st.title('DDPMPipeline Image Generator')
 
 if st.button('Generate Image'):
-    image = generate_image()
-    st.image(image, caption='Generated Image', use_column_width=True)
+    with st.spinner('Generating Image...'):
+        # Create a progress bar
+        progress_bar = st.progress(0)
+
+        for i in range(100):
+            # Update the progress bar with each iteration.
+            time.sleep(0.01)
+            progress_bar.progress(i + 1)
+
+        image = generate_image()
+        st.image(image, caption='Generated Image', use_column_width=True)