Update app.py

app.py

@@ -382,13 +382,14 @@ def run_client():
                     if selection_list == "Top-k DB in Taxonomy Intervention" or selection_list == "Top-k DB in Taxonomy Outcome":
                         response = relik_model.retrieve(text, k=50, batch_size=400, progress_bar=False)
                         candidates_text = [pred.document.text for pred in response[0] if pred.document.text in db_set]
+                        candidates_text = candidates_text[:10]
                     else:
-                        response = relik_model.retrieve(text, k=10, batch_size=400, progress_bar=False)
-                        candidates_text = [pred.document.text for pred in response[0]]
-    
+                        response = relik_model.retrieve(text, k=20, batch_size=400, progress_bar=False)
+                        candidates_text = [pred.document.text for pred in response[0]] 
+                        
         
                     if candidates_text:
-                        st.session_state.candidates = candidates_text[:10]
+                        st.session_state.candidates = candidates_text
 
                     else:
                         st.session_state.candidates = []