Update

app.py

@@ -1,6 +1,9 @@
 import json
 import time
 import streamlit as st
+from annotated_text import annotated_text
+
+import matplotlib
 
 from flair.data import Sentence
 from flair.models import SequenceTagger
@@ -9,11 +12,11 @@ checkpoints = [
     "qanastek/pos-french",
 ]
 
+colors = list(matplotlib.colors.cnames.values())
+
 @st.cache(suppress_st_warning=True, allow_output_mutation=True)
 def get_model(model_name):
-    
-    # Load the model
-    return SequenceTagger.load(model_name)
+    return SequenceTagger.load(model_name) # Load the model
 
 def getPos(s: Sentence):
     texts = []
@@ -27,6 +30,9 @@ def getPos(s: Sentence):
 def getDictFromPOS(texts, labels):
     return [{ "text": t, "label": l } for t, l in zip(texts, labels)]
 
+def getAnnotatedFromPOS(texts, labels):
+    return [(t,l,"#8ef") for t, l in zip(texts, labels)]
+
 def main():
 
     st.title("🥖 French-Part-Of-Speech-Tagging")
@@ -57,6 +63,10 @@ def main():
                 
                 st.header("Labels:")
                 st.write(" ".join(labels))
+                
+                st.header("Labels:")
+                anns = getAnnotatedFromPOS(texts, labels)
+                annotated_text(*anns)
 
                 st.header("JSON:")
                 st.json(getDictFromPOS(texts, labels))

requirements.txt

@@ -1 +1,3 @@
-flair==0.8.0.post1
+flair==0.8.0.post1
+st-annotated-text
+matplotlib