adding helper functions

app.py

@@ -15,9 +15,14 @@ def load_model(model_name):
     refined_model = Refined.from_pretrained(model_name=model_name, entity_set="wikipedia")
     return refined_model
 
-# Use the cache model
+# Use the cached model
 refined_model = load_model(selected_model_name)
 
+# Helper functions
+def get_wikidata_id(entity_string):
+  entity_list = entity_string.split("=")
+  return  "https://www.wikidata.org/wiki/" + str(entity_list[1])
+
 # Create the form
 with st.form(key='my_form'):
     text_input = st.text_input(label='Enter a sentence')