Update app.py

app.py

@@ -177,8 +177,8 @@ def load_model():
     )
 
     retriever_question_db = GoldenRetriever(
-        question_encoder="/home/user/app/models/retriever/question_encoder",
-        document_index="/home/user/app/models/retriever/document_index/question_db"
+        question_encoder="/home/user/app/models/retriever/level-4-small-no-negatives/question_encoder",
+        document_index="/home/user/app/models/retriever/level-4-small-no-negatives/document_index"
 
     )
 
@@ -283,9 +283,12 @@ def run_client():
 
         elif  analysis_type == "db intervention":
             relik_model = st.session_state["relik_model"][4]
+            
+        elif  analysis_type == "db outcome":
+            relik_model = st.session_state["relik_model"][5]
 
         else:
-            relik_model = st.session_state["relik_model"][0]
+            relik_model = st.session_state["relik_model"][3]
             
         text = text.strip()
         if text: