Update

app.py

@@ -3,8 +3,7 @@ import streamlit as st
 
 from flair.data import Sentence
 from flair.models import SequenceTagger
-
-from annotated_text import annotated_text
+from flair.visual.ner_html import render_ner_html
 
 checkpoints = [
     "qanastek/pos-french",
@@ -36,18 +35,16 @@ def main():
         with st.spinner("Search for Part-Of-Speech Tags 🔍"):
             
             # Build Sentence
-            sentence = Sentence(input_text)
+            s = Sentence(input_text)
 
             # predict tags
-            model.predict(sentence)
+            model.predict(s)
 
-            # print predicted pos tags
-            result = sentence.to_tagged_string()
+            # # print predicted pos tags
+            # result = sentence.to_tagged_string()
 
             try:
-                annotated_text(result)
-                st.write("")
-                st.success(result)
+                st.write(render_ner_html(sentences=[s], wrap_page=False), unsafe_allow_html=True)
             except Exception as e:
                 st.error("Some error occured!" + str(e))
                 st.stop()

requirements.txt

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