Chris4K commited on
Commit
1c5f1b8
·
verified ·
1 Parent(s): 5a3e5ee

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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) > 1:
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)])