Update app.py

app.py

@@ -100,7 +100,7 @@ def parse_text(text, prediction):
     parsed_text = []
     for p in prediction:
         parsed_text.append(text[start:p["start"]])
-        parsed_text.append((p["word"], p["entity_group"], ENTITY_TO_COLOR[p["entity_group"], "#000"]))
+        parsed_text.append((p["word"], p["entity_group"], ENTITY_TO_COLOR.get(p["entity_group"], "#000")))
         start = p["end"]
     parsed_text.append(text[start:])
     return parsed_text