"updated app.py"

app.py

@@ -3,7 +3,7 @@ from datasets import load_dataset
 import gradio as gr, json, os, random, torch
 from diffusers import StableDiffusionPipeline
 from gradio_client import Client
-from live_preview_helpers import sd_live_preview
+from live_preview_helpers import flux_pipe_call_that_returns_an_iterable_of_image
 
 def infer_sd_live(character):
     prompt = character.get("appearance", "portrait, character, digital art")

live_preview_helper.py

@@ -36,3 +36,29 @@ def sd_live_preview(pipe, prompt, num_inference_steps=30, width=512, height=512,
             )
             image = result.images[0]
         yield image
+# live_preview_helpers.py
+
+def flux_pipe_call_that_returns_an_iterable_of_images(
+    self,
+    prompt,
+    guidance_scale=7.5,
+    num_inference_steps=20,
+    width=512,
+    height=512,
+    generator=None,
+    output_type="pil",
+    good_vae=None
+):
+    """
+    Yields a single Stable Diffusion image matching the prompt.
+    """
+    image = self(
+        prompt=prompt,
+        guidance_scale=guidance_scale,
+        num_inference_steps=num_inference_steps,
+        width=width,
+        height=height,
+        generator=generator,
+        output_type=output_type,
+    ).images[0]
+    yield image