Update main.py
Browse files
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('
|
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')
|