Update app.py

app.py

@@ -373,7 +373,14 @@ def run_lora(prompt, cfg_scale, steps, selected_indices, lora_scale_1, lora_scal
             step_counter += 1
             images.append(image)
             progress_bar = f'<div class="progress-container"><div class="progress-bar" style="--current: {step_counter}; --total: {steps};"></div></div>'
-            yield images, seed, gr.update(value=progress_bar, visible=True)
+            if len(images) == 4:  # Ensure four images are collected
+                break
+
+        # Pad to ensure exactly 4 images are returned
+        while len(images) < 4:
+            images.append(None)
+
+        return *images, seed, gr.update(value=progress_bar, visible=True)
             
 run_lora.zerogpu = True