Update app.py

app.py

@@ -158,23 +158,6 @@ with open("/home/user/app/models/retriever/document_index/documents.jsonl", "r")
 @st.cache_resource()
 def load_model():
 
-    retriever = GoldenRetriever(
-        question_encoder="/home/user/app/models/retriever/question_encoder",
-        document_index=InMemoryDocumentIndex(
-                documents=DocumentStore.from_file(
-                    "/home/user/app/models/retriever/document_index/documents.jsonl"
-                ),
-                metadata_fields=["definition"],
-                separator=' <def> ',
-                device="cpu"
-            ),
-            devide="cpu"
-
-    )
-    retriever.index()
-
-
-
     retriever_question = GoldenRetriever(
         question_encoder="/home/user/app/models/retriever/question_encoder",
         document_index="home/user/app/models/retriever/document_index/questions"
@@ -287,9 +270,12 @@ def run_client():
 
     # ReLik API call
     if submit:
+        entity_linking_bool = False
+
+
         if analysis_type == "question":
             relik_model = st.session_state["relik_model"][0]
-
+            entity_linking_bool = True
         elif analysis_type == "intervention":
             relik_model = st.session_state["relik_model"][1]
         elif  analysis_type == "outcome":
@@ -310,7 +296,6 @@ def run_client():
         text = text.strip()
         if text:
             st.markdown("####")
-            st.markdown("#### Entity Linking")
             with st.spinner(text="In progress"):
                 response = relik_model(text)
 
@@ -324,15 +309,18 @@ def run_client():
                 dict_of_ents, options = get_el_annotations(response=response)
                 dict_of_ents_candidates, options_candidates = get_retriever_annotations(response=response)
 
-                display = displacy.render(
-                    dict_of_ents, manual=True, style="ent", options=options
-                )
-
-
-                display = display.replace("\n", " ")
+                if entity_linking_bool: 
+                    st.markdown("#### Entity Linking")
 
-                # heurstic, prevents split of annotation decorations
-                display = display.replace("border-radius: 0.35em;", "border-radius: 0.35em; white-space: nowrap;")
+                    display = displacy.render(
+                        dict_of_ents, manual=True, style="ent", options=options
+                    )
+    
+    
+                    display = display.replace("\n", " ")
+    
+                    # heurstic, prevents split of annotation decorations
+                    display = display.replace("border-radius: 0.35em;", "border-radius: 0.35em; white-space: nowrap;")
 
                 with st.container():
                     st.write(display, unsafe_allow_html=True)