Update app.py

app.py

@@ -41,7 +41,7 @@ torch.cuda.empty_cache()
 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 2048
 
-pipe.flux_pipe_call_that_returns_an_iterable_of_images = flux_pipe_call_that_returns_an_iterable_of_images.__get__(pipe)
+#pipe.flux_pipe_call_that_returns_an_iterable_of_images = flux_pipe_call_that_returns_an_iterable_of_images.__get__(pipe)
 
 @spaces.GPU
 def infer_flux(prompt, seed=42, randomize_seed=False, width=1024, height=1024, guidance_scale=3.5, num_inference_steps=28, progress=gr.Progress(track_tqdm=True)):
@@ -49,7 +49,7 @@ def infer_flux(prompt, seed=42, randomize_seed=False, width=1024, height=1024, g
         seed = random.randint(0, MAX_SEED)
     generator = torch.Generator().manual_seed(seed)
     
-    for img in pipe.flux_pipe_call_that_returns_an_iterable_of_images(
+    img = pipe(
             prompt=prompt,
             guidance_scale=guidance_scale,
             num_inference_steps=num_inference_steps,
@@ -58,8 +58,10 @@ def infer_flux(prompt, seed=42, randomize_seed=False, width=1024, height=1024, g
             generator=generator,
             output_type="pil",
             good_vae=good_vae,
-        ):
-            yield img
+        ).images[0]
+
+    img.save("flux-dev.png")
+    return "flux-dev.png"
 
 @spaces.GPU
 def llama_gen_fragrance(scene):