Batch now generates zip

app.py

@@ -12,6 +12,7 @@ import torch.nn.functional as F
 import matplotlib.pyplot as plt
 import warnings
 import tempfile
+from zipfile import ZipFile
 
 warnings.filterwarnings("ignore")
 
@@ -152,8 +153,13 @@ def bw_single(image_file):
 def batch(image_files):
     output = []
     for idx, file in enumerate(image_files):
-        output.append(inference(file.name))
-    return output
+        file = inference(file.name)
+        output.append(file)
+    
+    with ZipFile("tmp.zip", "w") as zipObj:
+        for idx, file in enumerate(output):
+            zipObj.write(file, file.split("/")[-1])
+    return output,"tmp.zip"
 
 with gr.Blocks() as iface:
     gr.Markdown("# Remove Background")
@@ -177,6 +183,8 @@ with gr.Blocks() as iface:
                 images = gr.File(file_count="multiple", file_types=["image"])
             with gr.Column():
                 gallery = gr.Gallery()
+                file_list = gr.Files(interactive=False)
+
         with gr.Row():
             with gr.Column():
                 batch_removebg = gr.Button("Batch Process")
@@ -184,7 +192,7 @@ with gr.Blocks() as iface:
                 batch_clear = gr.Button("Clear")
     #Events
     single_removebg.click(inference, inputs=image, outputs=image_output)
-    batch_removebg.click(batch, inputs=images, outputs=gallery)
+    batch_removebg.click(batch, inputs=images, outputs=[gallery,file_list])
     single_clear.click(lambda: None, None, image, queue=False)
     batch_clear.click(lambda: None, None, images, queue=False)