Update app.py

app.py

@@ -4,14 +4,13 @@ import json
 import requests
 
 import streamlit
-import spacy_streamlit
 import spacy
 from lxml import etree
 import pandas as pd
 
 # Constants
 CONFIG_FILE = "config.json"
-ASSETS_DIR = "./assets"
+ASSETS_DIR = "assets"
 XML_PARSER_CONFIG = {'ns_clean': True, 'recover': True, 'encoding': 'utf-8'}
 ENTITY_COLORS = {
     "EVENT": "#ec7063",
@@ -105,7 +104,7 @@ def main():
     entities = []
     # 1. User provides a XML EAD
     streamlit.write("## 📄 Input XML EAD:")
-    filename = streamlit.file_uploader("Upload an XML EAD (format .xml)", type="xml")
+    filename = streamlit.file_uploader("Upload an XML EAD (format .xml)", type="xml", label_visibility="collapsed")
     streamlit.markdown(
         "or use an XML EAD provided in [`samples/`](https://huggingface.co/spaces/ner4archives/ner4archives-NEL-vizualizer-app/blob/main/samples/) directory")
 
@@ -220,21 +219,21 @@ def main():
                 streamlit.write(df)
 
                 streamlit.write("## 🔎 Explore named entities in text: ")
-                spacy_streamlit.visualize_ner(
-                    {"text": plain,
+                ents_html = spacy.displacy.render(
+                    [{"text": plain,
                      "ents": [{"start": ent[0],
                                "end": ent[1],
                                "label": ent[3],
                                "kb_id": ent[4] if linking else "",
                                "kb_url": ent[5] if linking else ""
-                               } for ent in entities]},
-                    labels=["EVENT", "LOCATION", "ORGANISATION", "PERSON", "TITLE", 'LOC', 'MISC', 'ORG', 'PER'],
-                    show_table=False,
+                               } for ent in entities]}],
+                    style="ent",
                     manual=True,
-                    title="",
-                    displacy_options={
+                    options={
+                        "ents":["EVENT", "LOCATION", "ORGANISATION", "PERSON", "TITLE", 'LOC', 'MISC', 'ORG', 'PER'],
                         "colors": ENTITY_COLORS
                     })
+                streamlit.markdown(ents_html, unsafe_allow_html=True)
 
 
 if __name__ == "__main__":