Update

app.py

@@ -3,7 +3,7 @@ import random
 import streamlit as st
 from annotated_text import annotated_text
 
-import matplotlib
+from colour import Color
 
 from flair.data import Sentence
 from flair.models import SequenceTagger
@@ -12,8 +12,6 @@ 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):
     return SequenceTagger.load(model_name) # Load the model
@@ -30,8 +28,13 @@ def getPos(s: Sentence):
 def getDictFromPOS(texts, labels):
     return [{ "text": t, "label": l } for t, l in zip(texts, labels)]
 
+def randomColor():
+    rgb = (random.uniform(0.0,1.0), random.uniform(0.0,1.0), random.uniform(0.0,1.0))
+    hsl = (random.uniform(0.0,1.0), random.uniform(0.35,1.0), 0.62)
+    return str(Color(rgb=rgb, hsl=hsl))
+
 def getAnnotatedFromPOS(texts, labels):
-    return [(t,l,random.choice(colors)) for t, l in zip(texts, labels)]
+    return [(t,l,randomColor()) for t, l in zip(texts, labels)]
 
 def main():
 

requirements.txt

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