def get_id2label(labels):
    """
    Get id2label mapping based on labels

    Args:
      labels: list of labels.

    Return:
      id2label map
    """
    return {str(k): v for k, v in enumerate(labels)}