Update app.py

app.py

@@ -39,8 +39,8 @@ def chat_interface(inputs):
     
     #result = llm_chain({"input": query, "history": chat_history_tuples})
     result = llm_chain.predict(input = inputs)
-    formated_result = result.split('\n')[-1]
-    return formated_result
+    formated_result = result.split('AI:')[-1]
+    return formated_result.strip()
 
 llm = load_pipeline()
 chat_history = []