Spaces:
Sleeping
Sleeping
Commit
·
bebffa2
1
Parent(s):
f14ec94
changed num results
Browse files
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=
|
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 |
)
|