anasmkh commited on
Commit
9a529b4
·
verified ·
1 Parent(s): 89c77c3

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -49,7 +49,7 @@ qa_chain = RetrievalQA.from_llm(llm=local_llm, retriever=retriever)
49
 
50
  def gradinterface(query,history):
51
  result = qa_chain({'query': query})
52
- return result['result'].split(' : ')[-1].strip()
53
 
54
 
55
  demo = gr.ChatInterface(fn=gradinterface, title='OUR_OWN_BOT')
 
49
 
50
  def gradinterface(query,history):
51
  result = qa_chain({'query': query})
52
+ return result['result'].split(': ')[-1].strip()
53
 
54
 
55
  demo = gr.ChatInterface(fn=gradinterface, title='OUR_OWN_BOT')