Update app.py

app.py

@@ -203,8 +203,6 @@ def load_model():
 
     )
 
-
-
     
     retriever_outcome_gpt_taxonomy = GoldenRetriever(
         question_encoder="models/retriever/outcome/gpt/taxonomy/question_encoder",
@@ -263,7 +261,7 @@ def run_client():
         index=0  # Default to 'question'
     )
 
-    selection_options = ["DB Intervention", "DB Outcome", "Top-k DB Intervention", "Top-k Soft DB Outcome", "Taxonomy Intervention", "Taxonomy Outcome"]
+    selection_options = ["DB Intervention", "DB Outcome", "Taxonomy Intervention", "Taxonomy Outcome", "Top-k DB Intervention", "Top-k Soft DB Outcome", ]
     
     if analysis_type == "Retriever":
         # Selection list using selectbox
@@ -307,7 +305,10 @@ def run_client():
             entity_linking_bool = True
         else:
             model_idx = selection_options.index(selection_list)
-            relik_model = st.session_state["relik_model"][model_idx+1]
+            if "Top-k" in selection_options:
+                relik_model = st.session_state["relik_model"][model_idx-1]
+            else:
+                relik_model = st.session_state["relik_model"][model_idx+1]
             
             
         text = text.strip()