Update app.py

app.py

@@ -5,19 +5,22 @@ import numpy as np
 
 fawkes_protection = FawkesProtection()
 
-def process_image(image, protection_level):
-    protected_image = fawkes_protection.predict(image, level=protection_level)
-    return Image.fromarray(protected_image)
+def process_image(images, protection_level):
+    protected_images = []
+    for image in images:
+        protected_image = fawkes_protection.predict(image, level=protection_level)
+        protected_images.append(protected_image)
+    return protected_images
 
 iface = gr.Interface(
     fn=process_image,
     inputs=[
-        gr.Image(type="pil"),
+        gr.Image(type="pil", label="Images", elem_id="upload_images", multiupload=True),
         gr.Radio(["low", "mid", "high"], label="Protection Level")
     ],
-    outputs=gr.Image(type="pil"),
+    outputs=gr.Gallery(type="pil", label="Protected Images"),
     title="Fawkes Image Protection",
-    description="Upload an image to apply Fawkes protection."
+    description="Upload one or more images to apply Fawkes protection."
 )
 
 iface.launch()