Rams901 commited on
Commit
c40a87c
1 Parent(s): 5cf3cf7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -210,11 +210,12 @@ def qa_retrieve(query,):
210
  llm = ClaudeLLM2()
211
  chain = LLMChain(llm=llm, prompt = prompt)
212
 
213
- consensus = chain.run(query=query, articles="\n".join(tier_1), session_prompt = session_prompt,).strip()
214
-
215
  consensus = chain.run(query=query, articles="\n".join(tier_1), session_prompt = session_prompt,).strip()
216
  if "In conclusion" in consensus:
217
  consensus = consensus[:consensus.index('In conclusion')]
 
 
 
218
  intro = qa_intro(query, consensus, tier_1)
219
  conclusion = qa_conclusion(query, consensus, tier_1)
220
  cynthesis = intro + "\n\n" + consensus + conclusion
 
210
  llm = ClaudeLLM2()
211
  chain = LLMChain(llm=llm, prompt = prompt)
212
 
 
 
213
  consensus = chain.run(query=query, articles="\n".join(tier_1), session_prompt = session_prompt,).strip()
214
  if "In conclusion" in consensus:
215
  consensus = consensus[:consensus.index('In conclusion')]
216
+
217
+ consensus = consensus[consensus.index(':')+1:].strip()
218
+
219
  intro = qa_intro(query, consensus, tier_1)
220
  conclusion = qa_conclusion(query, consensus, tier_1)
221
  cynthesis = intro + "\n\n" + consensus + conclusion