Update app.py

app.py

@@ -9,7 +9,6 @@ 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"
@@ -44,24 +43,6 @@ 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))
 
@@ -78,7 +59,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", postprocess=save_image_with_fallback),  # Display multiple images in a gallery
+    outputs=gr.Gallery(label="Detected Persons"),  # Removed the postprocess argument
     examples=examples,
     cache_examples=False,  # Disable caching of examples
     examples_per_page=30,