luiscgp commited on
Commit
7457adf
1 Parent(s): 515946d
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -178,7 +178,7 @@ def start_haystack():
178
  return pipe
179
 
180
  @st.cache_resource
181
- def check_statement(statement: str, retriever_top_k: int = 5):
182
  """Run query and verify statement"""
183
  params = {"retriever": {"top_k": retriever_top_k}}
184
  return pipe.run(statement, params=params)
 
178
  return pipe
179
 
180
  @st.cache_resource
181
+ def check_statement(pipe, statement: str, retriever_top_k: int = 5):
182
  """Run query and verify statement"""
183
  params = {"retriever": {"top_k": retriever_top_k}}
184
  return pipe.run(statement, params=params)