Update

app.py

@@ -1,9 +1,9 @@
+import json
 import time
 import streamlit as st
 
 from flair.data import Sentence
 from flair.models import SequenceTagger
-from flair.visual.ner_html import render_ner_html
 
 checkpoints = [
     "qanastek/pos-french",
@@ -15,6 +15,17 @@ def get_model(model_name):
     # Load the model
     return SequenceTagger.load(model_name)
 
+def getPos(s: Sentence):
+    texts = []
+    labels = []
+    for t in s.tokens:
+        for label in t.annotation_layers.keys():
+            texts.append(t.text)
+            labels.append(t.get_labels(label)[0].value)          
+    return texts, labels
+
+def getDictFromPOS(texts, labels):
+    return [{ "text": t, "label": l } for t, l in zip(texts, labels)]
 
 def main():
 
@@ -40,16 +51,22 @@ def main():
             # predict tags
             model.predict(s)
 
-            # # print predicted pos tags
-            # result = sentence.to_tagged_string()
-
             try:
-                st.write(render_ner_html(sentences=[s], wrap_page=False), unsafe_allow_html=True)
+
+                texts, labels = getPos(s)
+                
+                st.write("Labels:")
+                st.write(" ".join(labels))
+
+                st.write("JSON:")
+                st.write(json.dumps(getDictFromPOS(texts, labels), indent=4))
+
             except Exception as e:
                 st.error("Some error occured!" + str(e))
                 st.stop()
 
     st.write("---")
+
     st.markdown(
         "Built by [Yanis Labrak](https://www.linkedin.com/in/yanis-labrak-8a7412145/) 🚀"
     )