Update app.py

app.py

@@ -6,11 +6,18 @@ from concurrent.futures import ThreadPoolExecutor
 
 fawkes_protection = FawkesProtection()
 
+def process_image(file, protection_level):
+    image = Image.open(file)
+    protected_image = fawkes_protection.predict(image, protection_level)
+    return Image.fromarray(protected_image)
+
 def process_images(image_files, protection_level):
-    images = [Image.open(file) for file in image_files]
+    results = []
     with ThreadPoolExecutor() as executor:
-        futures = [executor.submit(fawkes_protection.predict, img, protection_level) for img in images]
-        results = [Image.fromarray(future.result()) for future in futures]
+        futures = [executor.submit(process_image, file, protection_level) for file in image_files]
+        for future in futures:
+            result = future.result()
+            results.append(result)
     return results
 
 iface = gr.Interface(