Update app.py

app.py

@@ -38,19 +38,30 @@ def generate_images(
         # Set the track_tqdm parameter based on the number of images
         if num_images == 1:
             progress = gr.Progress(track_tqdm=True)
+            for _ in range(num_images):
+                output = pipe(
+                    prompt,
+                    negative_prompt=negative_prompt,
+                    num_inference_steps=num_inference_steps,
+                    guidance_scale=guidance_scale,
+                    height=height,
+                    width=width
+                )["images"][0]
+                outputs.append(output)
         else:
             progress = gr.Progress(track_tqdm=False)
+            for _ in progress.tqdm(range(num_images), desc="Generating images"):
+                output = pipe(
+                    prompt,
+                    negative_prompt=negative_prompt,
+                    num_inference_steps=num_inference_steps,
+                    guidance_scale=guidance_scale,
+                    height=height,
+                    width=width
+                )["images"][0]
+                outputs.append(output)
 
-        for _ in progress.tqdm(range(num_images), desc="Generating images"):
-            output = pipe(
-                prompt,
-                negative_prompt=negative_prompt,
-                num_inference_steps=num_inference_steps,
-                guidance_scale=guidance_scale,
-                height=height,
-                width=width
-            )["images"][0]
-            outputs.append(output)
+        
 
         return outputs
     else: