Update app.py

app.py

@@ -44,24 +44,20 @@ def extract_persons(image: np.ndarray, bboxes: np.ndarray) -> list[np.ndarray]:
     return person_images
 
 
-def convert_to_png(image: np.ndarray) -> bytes:
-    """Convert a NumPy image array to a PNG byte stream."""
-    pil_image = Image.fromarray(image)
-    buffer = io.BytesIO()
-    pil_image.save(buffer, format="PNG")
-    buffer.seek(0)
-    return buffer.read()
+def convert_to_pil_image(image: np.ndarray) -> Image.Image:
+    """Convert a NumPy image array to a PIL Image."""
+    return Image.fromarray(image)
 
 
 detector = load_model()
 detector.prepare(-1, nms_thresh=0.5, input_size=(640, 640))
 
 
-def detect(image: np.ndarray) -> list[bytes]:
+def detect(image: np.ndarray) -> list[Image.Image]:
     image = image[:, :, ::-1]  # RGB -> BGR
     bboxes = detect_person(image, detector)
     person_images = extract_persons(image, bboxes)  # Extract each person as a separate image
-    return [convert_to_png(person_img[:, :, ::-1]) for person_img in person_images]  # BGR -> RGB
+    return [convert_to_pil_image(person_img[:, :, ::-1]) for person_img in person_images]  # BGR -> RGB
 
 
 examples = sorted(pathlib.Path("images").glob("*.jpg"))