Update app.py

app.py

@@ -61,26 +61,26 @@ def get_document_store():
 document_store = get_document_store()
 
 
-@st.cache(allow_output_mutation=True)
-def get_retriever(document_store):
-    retriever = EmbeddingRetriever(
-        document_store=document_store,
-        embedding_model="sentence-transformers/multi-qa-mpnet-base-dot-v1",
-        model_format="sentence_transformers",
-        progress_bar=False,
-    )
-    return retriever
-
-retriever = get_retriever(document_store)
-
-
-# Instantiate retriever
-# retriever = EmbeddingRetriever(
-#   document_store=document_store,
-#   embedding_model="sentence-transformers/multi-qa-mpnet-base-dot-v1",
-#   model_format="sentence_transformers",
-#   progress_bar=False,
-# )
+# @st.cache(allow_output_mutation=True)
+# def get_retriever(document_store):
+#     retriever = EmbeddingRetriever(
+#         document_store=document_store,
+#         embedding_model="sentence-transformers/multi-qa-mpnet-base-dot-v1",
+#         model_format="sentence_transformers",
+#         progress_bar=False,
+#     )
+#     return retriever
+
+# retriever = get_retriever(document_store)
+
+
+Instantiate retriever
+retriever = EmbeddingRetriever(
+  document_store=document_store,
+  embedding_model="sentence-transformers/multi-qa-mpnet-base-dot-v1",
+  model_format="sentence_transformers",
+  progress_bar=False,
+)
 
 prompt_template="Answer the given question using the following documents. \
 Formulate your answer in the style of an academic report. \