Update app.py

app.py

@@ -20,7 +20,7 @@ st.set_page_config(
 # Sidebar
 st.sidebar.image("logo-wordlift.png")
 language_options = {"German", "English"}
-selected_language = st.sidebar.selectbox("Select the Language", list(language_options))
+selected_language = st.sidebar.selectbox("Select the Language", list(language_options), index=0)
 
 # Based on selected language, display model and entity set options
 if selected_language != "German":
@@ -30,6 +30,18 @@ if selected_language != "German":
     # Select entity_set
     entity_set_options = {"wikidata", "wikipedia"}
     selected_entity_set = st.sidebar.selectbox("Select the Entity Set", list(entity_set_options))
+
+    # Add citation for Refined
+    citation = """
+    @inproceedings{ayoola-etal-2022-refined,
+    title = "{R}e{F}in{ED}: An Efficient Zero-shot-capable Approach to End-to-End Entity Linking",
+    author = "Tom Ayoola, Shubhi Tyagi, Joseph Fisher, Christos Christodoulopoulos, Andrea Pierleoni",
+    booktitle = "NAACL",
+    year = "2022"
+    }
+    """
+    with st.sidebar.expander('Citations'):
+        st.markdown(citation)
 else:
     selected_model_name = None
     selected_entity_set = None
@@ -50,18 +62,6 @@ def load_model(selected_language, model_name=None, entity_set=None):
 # Use the cached model
 model = load_model(selected_language, selected_model_name, selected_entity_set)
 
-# Addi citation
-citation = """
-@inproceedings{ayoola-etal-2022-refined,
-title = "{R}e{F}in{ED}: An Efficient Zero-shot-capable Approach to End-to-End Entity Linking",
-author = "Tom Ayoola, Shubhi Tyagi, Joseph Fisher, Christos Christodoulopoulos, Andrea Pierleoni",
-booktitle = "NAACL",
-year = "2022"
-}
-"""
-with st.sidebar.expander('Citations'):
-    st.markdown(citation)
-
 # Helper functions
 def get_wikidata_id(entity_string):
     entity_list = entity_string.split("=")