Update app.py

app.py

@@ -8,6 +8,7 @@ import huggingface_hub
 import insightface
 import numpy as np
 import onnxruntime as ort
+from PIL import Image
 
 TITLE = "insightface Person Detection"
 DESCRIPTION = "https://github.com/deepinsight/insightface/tree/master/examples/person_detection"
@@ -45,8 +46,9 @@ def visualize(image: np.ndarray, bboxes: np.ndarray, vbboxes: np.ndarray) -> lis
         x1, y1, x2, y2 = bbox
         person_img = image[y1:y2, x1:x2]
         
-        # Append the cropped person image
-        person_images.append(person_img)
+        # Convert numpy array to PIL Image and append
+        pil_img = Image.fromarray(person_img)
+        person_images.append(pil_img)
 
     return person_images
 
@@ -60,7 +62,9 @@ def detect(image: np.ndarray) -> list[np.ndarray]:
     image = image[:, :, ::-1]  # RGB -> BGR
     bboxes, vbboxes = detect_person(image, detector)
     person_images = visualize(image, bboxes, vbboxes)
-    return [img[:, :, ::-1] for img in person_images]  # BGR -> RGB
+    
+    # Convert PIL images to numpy arrays and return
+    return [np.array(img) for img in person_images]
 
 demo = gr.Interface(
     fn=detect,