journey continues

app.py

@@ -28,7 +28,7 @@ image = Image.open("./test0.jpeg")
 labels = dataset.features['ner_tags'].feature.names
 id2label = {v: k for v, k in enumerate(labels)}
 
-label_ints = np.random.randint(0, len(ImageColor.colormap.items()), 60)
+label_ints = np.random.randint(0, len(ImageColor.colormap.items()), 61)
 label_color_pil = [k for k,_ in ImageColor.colormap.items()]
 label_color = [label_color_pil[i] for i in label_ints]
 label2color = {}