config.json

{
  "architectures": ["EfficientNetB4"],
  "image_size": 300,
  "id2label": {
    "0": "Heart",
    "1": "Oblong",
    "2": "Oval",
    "3": "Round",
    "4": "Square"
  },
  "label2id": {
    "Heart": 0,
    "Oblong": 1,
    "Oval": 2,
    "Round": 3,
    "Square": 4
  }
}