Update app.py

app.py

@@ -25,7 +25,8 @@ def infer(prompt, seed=42, randomize_seed=False, width=1024, height=1024, guidan
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
     generator = torch.Generator().manual_seed(seed)
-    
+
+    imgs = []
     for img in pipe.flux_pipe_call_that_returns_an_iterable_of_images(
             prompt=prompt,
             guidance_scale=guidance_scale,
@@ -36,8 +37,10 @@ def infer(prompt, seed=42, randomize_seed=False, width=1024, height=1024, guidan
             output_type="pil",
             good_vae=good_vae,
         ):
+            imgs.append(img)
+            imgs[0].save("output.gif", save_all=True, append_images=images[1:], duration=500, loop=0)
             yield img, seed
-    
+   
 examples = [
     "a tiny astronaut hatching from an egg on the moon",
     "a cat holding a sign that says hello world",
@@ -72,6 +75,7 @@ with gr.Blocks(css=css) as demo:
             run_button = gr.Button("Run", scale=0)
         
         result = gr.Image(label="Result", show_label=False)
+        gif = gr.Image(label="GIF")
         
         with gr.Accordion("Advanced Settings", open=False):
             
@@ -128,7 +132,6 @@ with gr.Blocks(css=css) as demo:
             outputs = [result, seed],
             cache_examples="lazy"
         )
-
     gr.on(
         triggers=[run_button.click, prompt.submit],
         fn = infer,