models/class_indices.json

{
  "Heart": 0,
  "Oblong": 1,
  "Oval": 2,
  "Round": 3,
  "Square": 4
}