image_filename = re.sub(r"^.*[\/\\]([^\/\\]+)$", r"\1", image_path)

app.py

@@ -16,6 +16,7 @@ import gradio as gr
 import numpy as np
 import torch
 import random
+import re
 from datetime import datetime
 
 from PIL import Image
@@ -160,7 +161,8 @@ def export_images_to_zip(gallery) -> str:
     with zipfile.ZipFile(tmp_zip.name, "w", compression=zipfile.ZIP_DEFLATED) as zf:
         for i in range(len(gallery)):
             image_path = gallery[i]
-            zf.write(image_path, arcname="image_" + str(i + 1) + ".webp")
+            image_filename = re.sub(r"^.*[\/\\]([^\/\\]+)$", r"\1", image_path)
+            zf.write(image_path, arcname=image_filename)
 
     print(str(len(gallery)) + " images zipped")
     return tmp_zip.name