def normalize_payload_image()

in api_inference_community/validation.py [0:0]


def normalize_payload_image(bpayload: bytes) -> Tuple[Any, Dict]:
    from PIL import Image

    try:
        # We accept both binary image with mimetype
        # and {"inputs": base64encodedimage}
        data = json.loads(bpayload)
        image = data["image"] if "image" in data else data["inputs"]
        image_bytes = b64decode(image)
        img = Image.open(BytesIO(image_bytes))
        return img, data.get("parameters", {})
    except Exception:
        pass

    img = Image.open(BytesIO(bpayload))
    return img, {}