Update app.py

app.py

@@ -84,7 +84,7 @@ def adr_predict(x):
         start = entity['start']
         end = entity['end']
         word = entity['word'].replace("##", "")
-        color = entity_colors[entity['entity_group']]
+        color = entity_colors.get(entity['entity_group'], 'black')
         
         htext += f"{x[prev_end:start]}<mark style='background-color:{color};'>{word}</mark>"
         prev_end = end