NealCaren commited on
Commit
22bc47a
·
verified ·
1 Parent(s): 3034838

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -146,7 +146,7 @@ def generate_literature_review(thesis, style_notes, api_key, progress=gr.Progres
146
 
147
  # Second attempt
148
  progress(0.6, desc="Finding additional references...")
149
- context2 = search_similar(get_embeddings(naive_results), chunk_df, k=10)
150
  text_chunks = [c['text'] for c in context]
151
  combo_context = context + [c for c in context2 if c['text'] not in text_chunks]
152
  final_cites = extract_cites(combo_context)
 
146
 
147
  # Second attempt
148
  progress(0.6, desc="Finding additional references...")
149
+ context2 = search_similar(get_embeddings(naive_results), chunk_df, k=16)
150
  text_chunks = [c['text'] for c in context]
151
  combo_context = context + [c for c in context2 if c['text'] not in text_chunks]
152
  final_cites = extract_cites(combo_context)