Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -285,7 +285,7 @@ def calculate_statistics(results, search_time, vector_store, num_tokens, embeddi
|
|
285 |
"top_k": top_k,
|
286 |
}
|
287 |
|
288 |
-
if len(results) >
|
289 |
embeddings = [embedding_model.embed_query(doc.page_content) for doc in results]
|
290 |
pairwise_similarities = np.inner(embeddings, embeddings)
|
291 |
stats["result_diversity"] = 1 - np.mean(pairwise_similarities[np.triu_indices(len(embeddings), k=1)])
|
|
|
285 |
"top_k": top_k,
|
286 |
}
|
287 |
|
288 |
+
if len(results) > 1000:
|
289 |
embeddings = [embedding_model.embed_query(doc.page_content) for doc in results]
|
290 |
pairwise_similarities = np.inner(embeddings, embeddings)
|
291 |
stats["result_diversity"] = 1 - np.mean(pairwise_similarities[np.triu_indices(len(embeddings), k=1)])
|