Update app.py

app.py

@@ -33,4 +33,37 @@ def detect_person(
     return bboxes
 
 
-def extract
+def extract_persons(image: np.ndarray, bboxes: np.ndarray) -> list[np.ndarray]:
+    person_images = []
+    for bbox in bboxes:
+        x1, y1, x2, y2 = bbox
+        person_image = image[y1:y2, x1:x2]  # Crop the detected person
+        person_images.append(person_image)
+    return person_images
+
+
+detector = load_model()
+detector.prepare(-1, nms_thresh=0.5, input_size=(640, 640))
+
+
+def detect(image: np.ndarray) -> list[np.ndarray]:
+    image = image[:, :, ::-1]  # RGB -> BGR
+    bboxes = detect_person(image, detector)
+    person_images = extract_persons(image, bboxes)  # Extract each person as a separate image
+    return [person_img[:, :, ::-1] for person_img in 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="Detected Persons", type="numpy", shape=(None, None)),  # Multiple outputs
+    examples=examples,
+    examples_per_page=30,
+    title=TITLE,
+    description=DESCRIPTION,
+)
+
+if __name__ == "__main__":
+    demo.queue(max_size=10).launch()