Update main.py

main.py

@@ -60,6 +60,7 @@ def gradinterface(query,history):
   Use three sentences maximum and keep the answer as concise as possible.
   {context}
   Question: {question}
+  
   Helpful Answer:"""
 
   QA_CHAIN_PROMPT = PromptTemplate(
@@ -72,7 +73,7 @@ def gradinterface(query,history):
         chain_type_kwargs={"prompt": QA_CHAIN_PROMPT},
     )
   result = qa_chain({"query": query})
-  return result['result']
+  return result['result'].split(': ')[-1].strip()
 
 
 demo = gr.ChatInterface(fn=gradinterface, title='OUR_OWN_BOT')