fix bug

app.py

@@ -227,6 +227,7 @@ def generate_image(
         neg_prompt = negative_prompt
 
     images = []
+    image_paths = []
     for s in seeds:
         generator = torch.Generator("cuda").manual_seed(int(s))
         image = pipe(
@@ -239,21 +240,23 @@ def generate_image(
             generator=generator,
         ).images[0]
         images.append(image)
+        tmp_img = tempfile.NamedTemporaryFile(delete=False, suffix=".png")
+        image.save(tmp_img.name, format="PNG")
+        image_paths.append(tmp_img.name)
 
-    return images, ", ".join(str(s) for s in seeds), images
+    return images, ", ".join(str(s) for s in seeds), image_paths
 
 
-def package_zip(images):
+def package_zip(image_paths):
     """Pack the current image list into a ZIP file for download."""
-    if not images:
+    if not image_paths:
         raise gr.Error("Please generate images first.")
 
     tmp = tempfile.NamedTemporaryFile(delete=False, suffix=".zip")
     with zipfile.ZipFile(tmp, "w", zipfile.ZIP_DEFLATED) as zf:
-        for idx, img in enumerate(images, start=1):
-            buf = io.BytesIO()
-            img.save(buf, format="PNG")
-            zf.writestr(f"image_{idx:02d}.png", buf.getvalue())
+        for idx, path in enumerate(image_paths, start=1):
+            # Store as image_01.png, image_02.png, ...
+            zf.write(path, arcname=f"image_{idx:02d}.png")
 
     tmp.flush()
     return tmp.name
@@ -374,7 +377,7 @@ with gr.Blocks(title="Z-Image-Turbo Demo", css=CUSTOM_CSS, analytics_enabled=Fal
     gr.Markdown(
         """
         <div class="footer-note">
-        Model: Tongyi-MAI/Z-Image-Turbo (Apache 2.0). Demo by mrfakename · FA3/AoTI tweaks by multimodalart.
+        Model: Tongyi-MAI/Z-Image-Turbo (Apache 2.0). Demo by https://z-image-turbo.tech · 
         </div>
         """
     )