Update app.py

app.py

@@ -24,6 +24,18 @@ def load_model(model_name, entity_set):
 # Use the cached model
 refined_model = load_model(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.beta_expander('Citations'):
+    st.markdown(citation)
+
 # Helper functions
 def get_wikidata_id(entity_string):
     entity_list = entity_string.split("=")