Upload streamlit_app.py

src/streamlit_app.py

@@ -80,8 +80,9 @@ def ner_with_robertime(text: str) -> List[Tuple[str, str]]:
 def colorize_entities(ner_result: List[Tuple[str, str]]) -> str:
     html = ''
     for token, label in ner_result:
-        color = LABEL_COLORS.get(label, '#eeeeee')
-        if label != 'LABEL-0':
+        norm_label = label.replace('_', '-')
+        if norm_label != 'LABEL-0':
+            color = LABEL_COLORS.get(norm_label, '#eeeeee')
             html += f'<span style="background-color:{color};padding:2px 4px;border-radius:4px;margin:1px;">{token}</span> '
         else:
             html += f'{token} '