Update app.py

app.py

@@ -161,6 +161,7 @@ def generate_image(setup_args, num_iterations):
                 if os.path.exists(final_image_path):
                     iter_images = list_iter_images(save_dir)
                     torch.cuda.empty_cache()  # Free up cached memory
+                    time.sleep(0.5)
                     yield (final_image_path, f"Final image saved at {final_image_path}", iter_images)
                 else:
                     torch.cuda.empty_cache()  # Free up cached memory