Update app.py

app.py

@@ -22,17 +22,15 @@ st.sidebar.image("logo-wordlift.png")
 language_options = {"English", "German"}
 selected_language = st.sidebar.selectbox("Select the Language", list(language_options), index=0)
 
-# Based on selected language, display model and entity set options
+# Based on selected language, configure model, entity set, and citation options
 if selected_language != "German":
-    # Only show these options for languages other than German
-    model_options = {"aida_model", "wikipedia_model_with_numbers"}
-    selected_model_name = st.sidebar.selectbox("Select the Model", list(model_options))
-    # 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 = """
+    model_options = ["aida_model", "wikipedia_model_with_numbers"]
+    entity_set_options = ["wikidata", "wikipedia"]
+    
+    selected_model_name = st.sidebar.selectbox("Select the Model", model_options)
+    selected_entity_set = st.sidebar.selectbox("Select the Entity Set", entity_set_options)
+
+    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",
@@ -40,14 +38,14 @@ if selected_language != "German":
     year = "2022"
     }
     """
+
     with st.sidebar.expander('Citations'):
-        st.markdown(citation)
+        st.markdown(refined_citation)
 else:
     selected_model_name = None
     selected_entity_set = None
 
-    # Add citation for entity fishing
-    citation = """
+    entity_fishing_citation = """
     @misc{entity-fishing,
     title = {entity-fishing},
     howpublished = {\url{https://github.com/kermitt2/entity-fishing}},
@@ -57,8 +55,9 @@ else:
     eprint = {1:dir:cb0ba3379413db12b0018b7c3af8d0d2d864139c}
     }
     """
+
     with st.sidebar.expander('Citations'):
-        st.markdown(citation)
+        st.markdown(entity_fishing_citation)
 
 @st.cache_resource  # 👈 Add the caching decorator
 def load_model(selected_language, model_name=None, entity_set=None):