Update app.py

app.py

@@ -37,7 +37,7 @@ pinecone_key = os.environ["PINECONE_API_KEY"]
 # )
 
 
-@st.cache(allow_output_mutation=True)
+@st.cache_resource 
 def get_document_store():
     # Assuming you're using Elasticsearch as a document store
     doc_file_name="cpv_full_southern_africa"
@@ -46,20 +46,27 @@ def get_document_store():
                                        index=doc_file_name)
     return document_store
 
-@st.cache(allow_output_mutation=True)
-def get_retriever(document_store):
-    # Initialize the Pinecone retriever configuration here
-    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
+# @st.cache(allow_output_mutation=True)
+# def get_retriever(document_store):
+#     # Initialize the Pinecone retriever configuration here
+#     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
 
 # Get (or initialize and get) the document store and retriever
 document_store = get_document_store()
-retriever = get_retriever(document_store)
+# retriever = 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,
+)
 
 # template = PromptTemplate(
 #     prompt="""