Update app.py

app.py

@@ -38,9 +38,11 @@ if text:
         elif result["entity"].startswith("LABEL_2"):
             result["entity"] = "I"
     
-    mendo =[]
+    mendo = []
+    spanMendo = []
     for result in formatted_results:
         if not result["entity"].startswith("O"):
+            spanMendo.append((result["start"],result["end"],result["entity"]))
             mendo.append(f"""Entity: {result["entity"]}, Start:{result["start"]}, End:{result["end"]}, word:{text[result["start"]:result["end"]]}""")        
     
     # Base Model     
@@ -70,10 +72,12 @@ if text:
             result["entity"] = "I"
     
     base=[]        
+    spanBase=[]
     for result in formatted_results:
         if not result["entity"].startswith("O"):
+            spanBase.append((result["start"],result["end"],result["entity"]))
             base.append(f"""Entity: {result["entity"]}, Start:{result["start"]}, End:{result["end"]}, word:{text[result["start"]:result["end"]]}""")
 
-    st.json(base)
-    st.json(mendo)
+    st.text(show_span_box_markup(text, spanMendo))
+    st.text(show_span_box_markup(text, spanBase))