add models in YAML

README.md

@@ -6,7 +6,8 @@ colorTo: blue
 sdk: streamlit
 sdk_version: 1.10.0
 app_file: Rock_fact_checker.py
-pinned: false
+pinned: true
+models: [sentence-transformers/msmarco-distilbert-base-tas-b, microsoft/deberta-v2-xlarge-mnli]
 license: apache-2.0
 ---
 

app_utils/backend_utils.py

@@ -56,7 +56,7 @@ pipe = start_haystack()
 
 # the pipeline is not included as parameter of the following function,
 # because it is difficult to cache
-@st.cache(persist=True, allow_output_mutation=True)
+@st.cache(allow_output_mutation=True)
 def query(statement: str, retriever_top_k: int = 5):
     """Run query and verify statement"""
     params = {"retriever": {"top_k": retriever_top_k}}