Adding chat history

app.py

@@ -26,8 +26,15 @@ tokenizer.padding_side = "right"
 pipe = pipeline("text-generation", model=model, tokenizer=tokenizer)
 
 def infer(message, history):
-    prompt = pipe.tokenizer.apply_chat_template([{"role": "user", "content": message}], tokenize=False, add_generation_prompt=True)
+    chat_list = []
+    for chat in history:
+        chat_user = {"role":"user", "content":chat[0]}
+        chat_assistant = {"role":"assistant", "content":chat[1]}
+        chat_list.append(chat_user)
+        chat_list.append(chat_assistant)
+    chat_list.append({"role": "user", "content": message})        
+    prompt = pipe.tokenizer.apply_chat_template(chat_list, tokenize=False, add_generation_prompt=True)
     outputs = pipe(prompt, max_new_tokens=256, do_sample=True, num_beams=1, temperature=0.3, top_k=50, top_p=0.95, max_time= 180)
     return outputs[0]['generated_text'][len(prompt):].strip()
 
-gr.ChatInterface(infer).launch()
+gr.ChatInterface(infer, title="Phi-3 Assistant").launch()