Update app.py

app.py

@@ -35,13 +35,13 @@ st.markdown('## NER-Visualisierer für deutsche (historische) Texte')
 
 #Spacy-Modell auswählen 
 model = "de_fnhd_nerdh"
-model = st.sidebar.selectbox("Wähle ein Modell:", ["de_fnhd_nerdh", "de_fnhd_sm", "de_fnhd_md", "de_fnhd_lg", "de_core_news_sm", "de_core_news_md", "de_core_news_lg"], key="model")
+model = st.sidebar.selectbox("Wähle ein Modell:", ["de_fnhd_nerdh", "de_core_news_sm", "de_core_news_md", "de_core_news_lg"], key="model")
 with st.spinner('Modell wird geladen...'):
     nlp = spacy.load(model)
 st.success('Modell ' + model + ' ist geladen!')
 
 #Informationen über die verschiedenen Modelle
-if model in ["de_fnhd_nerdh","de_fnhd_sm", "de_fnhd_md", "de_fnhd_lg"]:
+if model == "de_fnhd_nerdh":
     with st.sidebar.expander("Über das Modell"):
         st.write("""
             Das Modell basiert auf frühneuhochdeutschen Texten der digitalen Edition **Philipp Hainhofer: *Reiseberichte & Sammlungsbeschreibungen 1594-1636*** (https://hainhofer.hab.de/). 
@@ -136,14 +136,15 @@ st.success("Text ist eingelesen!")
 #------------------------------------------------------------------------------
 
 #Farben für die verschiedenen Entitäten 
-colors = {"PER": "#fdec3e", "PERSON": "#fdec3e", "LOC": "#7e56c2", "ORT": "#7e56c2", "ORG": "#209485" , "ORGANISATION": "#209485" , "MISC": "#eb4034",  "ZEIT": "#4c9c4b"}
-        #Spacy-Streamlit NER Visualizer
+colors = {"PER": "#fdec3e", "PERSON": "#fdec3e", "LOC": "#7e56c2", "ORT": "#7e56c2", "ORG": "#209485" , "ORGANISATION": "#209485" , "MISC": "#eb4034",  "ZEIT": "#4c9c4b", "OBJEKT": "#7e56c2"}
+
+#Spacy-Streamlit NER Visualizer
 
 #NER-Prozess wird gestartet, je nach Model werden hier die entsprechenden Entitäten gewechselt.
 with st.spinner('Named Entities werden gesucht...'):
     doc = nlp(text)
-    if model in ["de_fnhd_nerdh","de_fnhd_sm", "de_fnhd_md", "de_fnhd_lg"]:    
-        spacy_streamlit.visualize_ner(doc, labels = ["PERSON", "ORT", "ORGANISATION", "OBJEKT", "LITERATUR", "ZEIT",], show_table=False,    colors = colors)
+    if model == "de_fnhd_nerdh":    
+        spacy_streamlit.visualize_ner(doc, labels = ["PERSON", "ORT", "ORGANISATION", "OBJEKT", "ZEIT",], show_table=False,    colors = colors)
     else: 
         spacy_streamlit.visualize_ner(doc, labels = ["PER", "LOC", "ORG", "MISC"], show_table=False,    colors = colors)
 st.success('Suchprozess ist abgeschlossen!')