Update

app.py

@@ -1,12 +1,10 @@
 import time
-
-import errant
 import streamlit as st
 
 from flair.data import Sentence
 from flair.models import SequenceTagger
 
-from highlighter import show_highlights
+from annotated_text import annotated_text
 
 checkpoints = [
     "qanastek/pos-french",
@@ -18,16 +16,11 @@ def get_model(model_name):
     # Load the model
     return SequenceTagger.load(model_name)
 
-@st.cache(suppress_st_warning=True, allow_output_mutation=True)
-def get_annotator(lang: str):
-    return errant.load(lang)
-
 
 def main():
 
     st.title("🥖 French-Part-Of-Speech-Tagging")
 
-    annotator = get_annotator("fr")
     checkpoint = st.selectbox("Choose model", checkpoints)
     model = get_model(checkpoint)
 
@@ -52,7 +45,7 @@ def main():
             result = sentence.to_tagged_string()
 
             try:
-                show_highlights(annotator, input_text, result)
+                annotated_text(result)
                 st.write("")
                 st.success(result)
             except Exception as e:

requirements.txt

@@ -1,5 +1,4 @@
 flair==0.8.0.post1
 beautifulsoup4
 pandas
-st-annotated-text
-errant
+st-annotated-text