def decode_image()

in src/sagemaker_huggingface_inference_toolkit/decoder_encoder.py [0:0]

• 3 lines of code
• 1 McCabe index (conditional complexity)

def decode_image(bpayload: bytearray):
    """Convert a .jpeg / .png / .tiff... object to a proper inputs dict.
    Args:
        bpayload (bytes): byte stream.
    Returns:
        (dict): dictonatry for input
    """
    image = Image.open(BytesIO(bpayload)).convert("RGB")
    return {"inputs": image}