Update app.py

app.py

@@ -297,12 +297,9 @@ def run_client():
 
     # ReLik API call
     if submit:
-        entity_linking_bool = False
-
 
         if analysis_type == "Entity Linking":
             relik_model = st.session_state["relik_model"][0]
-            entity_linking_bool = True
         else:
             model_idx = selection_options.index(selection_list)
             if "Top-k" in selection_options:
@@ -315,7 +312,7 @@ def run_client():
         if text:
             st.markdown("####")
             with st.spinner(text="In progress"):
-                if entity_linking_bool:
+                if analysis_type == "Entity Linking":
                     response = relik_model(text)
                     
                     dict_of_ents, options = get_el_annotations(response=response)
@@ -343,7 +340,7 @@ def run_client():
     
                     st.markdown(text, unsafe_allow_html=True)
                 else:
-                    if "Top-k" in selection_options:
+                    if analysis_type == "Top-k DB in Taxonomy Intervention" or analysis_type == "Top-k DB in Taxonmy Outcome":
                         response = relik_model.retrieve(text, k=100, batch_size=400, progress_bar=False)
                            
                         candidates_text = []