Update app.py

app.py

@@ -61,13 +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,
-)
+# 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. \