Update app.py

app.py

@@ -1,5 +1,4 @@
 from __future__ import annotations
-import pathlib
 import io
 import cv2
 import gradio as gr
@@ -66,6 +65,9 @@ def extract_persons(image: np.ndarray, bboxes: np.ndarray) -> list[Image.Image]:
         x1, y1, x2, y2 = bbox
         person_image = image[y1:y2, x1:x2]  # Crop the detected person
         person_pil_image = Image.fromarray(person_image).convert('RGB')  # Convert to RGB
+        with io.BytesIO() as output:
+            person_pil_image.save(output, format='JPEG')  # Convert to JPEG
+            person_pil_image = Image.open(output)  # Reopen to ensure format
         person_images.append(person_pil_image)
     return person_images
 
@@ -82,14 +84,10 @@ def detect(image: np.ndarray) -> tuple[Image.Image, list[Image.Image]]:
     return Image.fromarray(res[:, :, ::-1], 'RGB'), person_images  # BGR -> RGB
 
 
-examples = sorted(pathlib.Path("images").glob("*.jpg"))
-
 demo = gr.Interface(
     fn=detect,
     inputs=gr.Image(label="Input", type="numpy"),
     outputs=[gr.Image(label="Processed Image", type="numpy"), gr.Gallery(label="Detected Persons", type="numpy")],
-    examples=examples,
-    examples_per_page=30,
     title=TITLE,
     description=DESCRIPTION,
 )