Update app.py

app.py

@@ -9,6 +9,7 @@ import insightface
 import numpy as np
 import onnxruntime as ort
 from PIL import Image
+import io
 
 TITLE = "insightface Person Detection"
 DESCRIPTION = "https://github.com/deepinsight/insightface/tree/master/examples/person_detection"
@@ -43,6 +44,24 @@ def extract_persons(image: np.ndarray, bboxes: np.ndarray) -> list[np.ndarray]:
     return person_images
 
 
+def encode_pil_to_bytes(img: Image.Image, format="WEBP", fallback_format="PNG") -> bytes:
+    output_bytes = io.BytesIO()
+    try:
+        img.save(output_bytes, format=format)
+    except Exception as e:
+        print(f"WebP save failed: {e}, falling back to {fallback_format}")
+        img.save(output_bytes, format=fallback_format)
+    return output_bytes.getvalue()
+
+
+def save_image_with_fallback(img: Image.Image, cache_dir: pathlib.Path, fallback_format="PNG") -> pathlib.Path:
+    try:
+        return gr.processing_utils.save_pil_to_cache(img, cache_dir, format="WEBP")
+    except Exception as e:
+        print(f"WebP save failed: {e}, falling back to {fallback_format}")
+        return gr.processing_utils.save_pil_to_cache(img, cache_dir, format=fallback_format)
+
+
 detector = load_model()
 detector.prepare(-1, nms_thresh=0.5, input_size=(640, 640))
 
@@ -59,7 +78,7 @@ examples = sorted(pathlib.Path("images").glob("*.jpg"))
 demo = gr.Interface(
     fn=detect,
     inputs=gr.Image(label="Input", type="numpy"),
-    outputs=gr.Gallery(label="Detected Persons"),  # Display multiple images in a gallery
+    outputs=gr.Gallery(label="Detected Persons", postprocess=save_image_with_fallback),  # Display multiple images in a gallery
     examples=examples,
     cache_examples=False,  # Disable caching of examples
     examples_per_page=30,