Update main.py

main.py

@@ -50,7 +50,7 @@ qa_chain =  RetrievalQA.from_llm(llm=local_llm, retriever=retriever)
 
 def gradinterface(query,history):
     result = qa_chain({'query': query})
-    return result['result'].split(': ')[-1].strip()
+    return result.split(' : ').strip()
 
 
 demo = gr.ChatInterface(fn=gradinterface, title='OUR_OWN_BOT')