Mikeplockhart commited on
Commit
e8c22b8
1 Parent(s): 7967a92

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -56,8 +56,8 @@ def reranking_results(query: str, top_k_results: list[str]):
56
  reranked_results = rerank_model.rank(query, top_k_results, return_documents=True)
57
  return reranked_results
58
 
59
- def run_query(query_string: str, collection):
60
- meal_string = search_chroma(collection, query_string)
61
  return meal_string
62
 
63
  with gr.Blocks() as meal_search:
 
56
  reranked_results = rerank_model.rank(query, top_k_results, return_documents=True)
57
  return reranked_results
58
 
59
+ def run_query(query_string: str):
60
+ meal_string = search_chroma(query_string)
61
  return meal_string
62
 
63
  with gr.Blocks() as meal_search: