Update app.py

app.py

@@ -5,6 +5,7 @@ import time
 
 import streamlit as st
 
+from stqdm import stqdm
 from diffusers import DiffusionPipeline, UNet2DConditionModel
 from PIL import Image
 
@@ -72,7 +73,7 @@ def main():
 
         generator = torch.Generator(device=device).manual_seed(seed)
         images = []
-        for img_idx in range(num_images):
+        for img_idx in stqdm(range(num_images)):
             generated_image = pipeline(prompt, num_inference_steps=30, generator=generator).images[0]
             images.append(generated_image)