make bbox parsing less strict

app.py

@@ -201,7 +201,7 @@ def perform_anonymisation(input_image_pil: Image.Image, raw_model_output: str) -
         print("Parsing findings...")
         # Use the provided utility functions
         parsed_findings = utils.parse_into_models(
-            utils.parse_json_response(raw_model_output)
+            utils.parse_json_response(raw_model_output), strict=False
         )
         print(f"[+] Parsed {len(parsed_findings)} findings.")
         if not parsed_findings:

utils.py

@@ -1,3 +1,4 @@
+from typing import Union
 from pydantic import BaseModel, field_validator
 import numpy as np
 import json
@@ -153,7 +154,7 @@ def parse_json_response(out: str) -> list[dict]:
     return json.loads(fixed)
 
 
-def parse_into_models(findings: list[dict]) -> list[Finding]:
+def parse_into_models(findings: list[dict], strict=True) -> Union[list[Finding], list[BoundingBox]]:
     """Parses and validates a list of dictionaries into a list of Finding models.
 
     Args:
@@ -162,6 +163,8 @@ def parse_into_models(findings: list[dict]) -> list[Finding]:
     Returns:
         A list of validated Finding model instances.
     """
+    if not strict:
+        return [BoundingBox.model_validate(box) for box in findings]
     return [Finding.model_validate(box) for box in findings]