Update app.py

app.py

@@ -38,8 +38,9 @@ def infer(prompt, seed=42, randomize_seed=False, width=1024, height=1024, guidan
             good_vae=good_vae,
         ):
             images.append(img)
-            images[0].save("output.gif", save_all=True, append_images=images[1:], duration=500, loop=0)
-            yield img, seed, "output.gif"
+        
+    images[0].save("output.gif", save_all=True, append_images=images[1:], duration=500, loop=0)
+    yield images[-1], seed, "output.gif"
    
 examples = [
     "a tiny astronaut hatching from an egg on the moon",