def METHOD_NAME(image, label):
    pred = _prediction_to_deepchecks_format(model, image)
    # class_id is required to be the first column
    label = tf.gather(label, [4, 0, 1, 2, 3], axis=1) if label is not None and len(label) > 0 else label
    return {'images': [image], 'labels': [label], 'predictions': [pred]}