Spaces:
Running
Running
anakin87
commited on
Commit
•
4c5a7dc
1
Parent(s):
7a393a7
add models in YAML
Browse files- README.md +2 -1
- app_utils/backend_utils.py +1 -1
README.md
CHANGED
@@ -6,7 +6,8 @@ colorTo: blue
|
|
6 |
sdk: streamlit
|
7 |
sdk_version: 1.10.0
|
8 |
app_file: Rock_fact_checker.py
|
9 |
-
pinned:
|
|
|
10 |
license: apache-2.0
|
11 |
---
|
12 |
|
|
|
6 |
sdk: streamlit
|
7 |
sdk_version: 1.10.0
|
8 |
app_file: Rock_fact_checker.py
|
9 |
+
pinned: true
|
10 |
+
models: [sentence-transformers/msmarco-distilbert-base-tas-b, microsoft/deberta-v2-xlarge-mnli]
|
11 |
license: apache-2.0
|
12 |
---
|
13 |
|
app_utils/backend_utils.py
CHANGED
@@ -56,7 +56,7 @@ pipe = start_haystack()
|
|
56 |
|
57 |
# the pipeline is not included as parameter of the following function,
|
58 |
# because it is difficult to cache
|
59 |
-
@st.cache(
|
60 |
def query(statement: str, retriever_top_k: int = 5):
|
61 |
"""Run query and verify statement"""
|
62 |
params = {"retriever": {"top_k": retriever_top_k}}
|
|
|
56 |
|
57 |
# the pipeline is not included as parameter of the following function,
|
58 |
# because it is difficult to cache
|
59 |
+
@st.cache(allow_output_mutation=True)
|
60 |
def query(statement: str, retriever_top_k: int = 5):
|
61 |
"""Run query and verify statement"""
|
62 |
params = {"retriever": {"top_k": retriever_top_k}}
|