Colorier les Tags autres que "O"

app.py

@@ -92,8 +92,16 @@ def tag_sentence(text):
     # Remplacez les étiquettes par des valeurs numériques
     df['tag'] = df['tag'].map(id2tag)
 
-    return df
+    # Appliquez une mise en forme conditionnelle pour colorier les tags dans le texte
+    def color_tags(tag):
+        if tag == 'O':
+            return ''
+        else:
+            return 'color: blue'
 
+    df['word'] = df.apply(lambda row: f'<span style="{color_tags(row["tag"])}">{row["word"]}</span>', axis=1)
+
+    return df
 
 st.title("📘 Named Entity Recognition Wolof")
 
@@ -129,7 +137,7 @@ if submit_button:
         c1, c2, c3 = st.columns([1, 3, 1])
 
         with c2:
-            st.table(results.style.background_gradient(subset=['tag']).format(precision=2))
+            st.write(results.to_html(escape=False), unsafe_allow_html=True)
 
 st.header("")
 st.header("")
@@ -142,4 +150,4 @@ with st.expander("ℹ️ - About this app", expanded=True):
 -   The app uses the [XLMRoberta model](https://huggingface.co/xlm-roberta-base), fine-tuned on the [masakhaNER](https://huggingface.co/datasets/masakhane/masakhaner2) dataset.
 -   The model uses the **byte-level BPE tokenizer**. Each sentence is first tokenized.
         """
-    )
+)