IMAGE_TFREC_KEY = "image"
IMAGE_SHAPE_TFREC_KEY = "image_shape"
LABEL_TFREC_KEY = "label"

MODEL_INPUT_IMAGE_KEY = "pixel_values"
MODEL_INPUT_LABEL_KEY = "labels"
IMAGE_MODEL_KEY = "pixel_values"
LABEL_MODEL_KEY = "labels"

CONCRETE_INPUT = "pixel_values"

PRETRAIN_CHECKPOINT = "google/vit-base-patch16-224-in21k"

LABELS = ["angular_leaf_spot", "bean_rust", "healthy"]
