Update app.py

app.py

@@ -462,7 +462,7 @@ def generate(
     filename = f"z_generated_{timestamp}_{seed}.png"
     tmp_path = os.path.join("/tmp", filename)
     os.makedirs("/tmp", exist_ok=True)
-    final_img.save(tmp_path)
+    image.save(tmp_path)
     
     return gallery_images, str(new_seed), int(new_seed), tmp_path