Update app.py

app.py

@@ -38,9 +38,12 @@ def chat_interface(inputs):
         chat_history_tuples.append((message[0], message[1]))
     
     #result = llm_chain({"input": query, "history": chat_history_tuples})
-    result = llm_chain.predict(input = inputs)
-    formated_result = result.split('AI:')[-1]
-    return formated_result.strip()
+    result = llm_chain.invoke({
+        "input" : inputs,
+        "history" : chat_history
+    })
+    formated_result = result #.split('AI:')[-1]
+    return formated_result #.strip()
 
 llm = load_pipeline()
 chat_history = []