Update app.py

app.py

@@ -302,7 +302,7 @@ def run_client():
             relik_model = st.session_state["relik_model"][0]
         else:
             model_idx = selection_options.index(selection_list)
-            if analysis_type == "Top-k DB in Taxonomy Intervention" or analysis_type == "Top-k DB in Taxonmy Outcome":
+            if selection_list == "Top-k DB in Taxonomy Intervention" or selection_list == "Top-k DB in Taxonmy Outcome":
                 relik_model = st.session_state["relik_model"][model_idx-1]
             else:
                 relik_model = st.session_state["relik_model"][model_idx+1]
@@ -340,7 +340,7 @@ def run_client():
     
                     st.markdown(text, unsafe_allow_html=True)
                 else:
-                    if analysis_type == "Top-k DB in Taxonomy Intervention" or analysis_type == "Top-k DB in Taxonmy Outcome":
+                    if selection_list == "Top-k DB in Taxonomy Intervention" or selection_list == "Top-k DB in Taxonmy Outcome":
                         response = relik_model.retrieve(text, k=100, batch_size=400, progress_bar=False)
                            
                         candidates_text = []