HonestAnnie commited on
Commit
bebffa2
1 Parent(s): f14ec94

changed num results

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -20,7 +20,7 @@ def get_embeddings(queries, task):
20
  def query_chroma(collection, embedding, authors):
21
  results = collection.query(
22
  query_embeddings=[embedding.tolist()],
23
- n_results=10,
24
  where={"author": {"$in": authors}} if authors else {},
25
  include=["documents", "metadatas", "distances"]
26
  )
 
20
  def query_chroma(collection, embedding, authors):
21
  results = collection.query(
22
  query_embeddings=[embedding.tolist()],
23
+ n_results=20,
24
  where={"author": {"$in": authors}} if authors else {},
25
  include=["documents", "metadatas", "distances"]
26
  )