Update app.py

app.py

@@ -30,7 +30,7 @@ Image.open(dataset[1]["image_path"]).convert("RGB").save("example2.png")
 Image.open(dataset[0]["image_path"]).convert("RGB").save("example3.png")
 # define id2label, label2color
 labels = dataset.features['ner_tags'].feature.names
-id2label = {v: k for v, k in enumerate(labels)}
+id2label = {0: 'O', 1: 'B-HEADER', 2: 'I-HEADER', 3: 'B-QUESTION', 4: 'I-QUESTION', 5: 'B-ANSWER', 6: 'I-ANSWER'}
 label2color = {'question':'blue', 'answer':'green', 'header':'orange', 'other':'violet'}