tjl8 commited on
Commit
2566043
·
verified ·
1 Parent(s): 1d25604

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -537,7 +537,7 @@ def semantic_search(query, embeddings, model, threshold=0.5):
537
  sims = cosine_similarity(query_embedding, embeddings)[0]
538
  return [(i, s) for i, s in enumerate(sims) if s > threshold]
539
 
540
- def rag_summarize(texts, summarizer, top_k=10):
541
  if not texts:
542
  return "No relevant content to summarize."
543
  vect = TfidfVectorizer()
 
537
  sims = cosine_similarity(query_embedding, embeddings)[0]
538
  return [(i, s) for i, s in enumerate(sims) if s > threshold]
539
 
540
+ def rag_summarize(texts, summarizer, top_k=5):
541
  if not texts:
542
  return "No relevant content to summarize."
543
  vect = TfidfVectorizer()