Update app.py

app.py

@@ -2,7 +2,7 @@ import os
 
 import gradio as gr
 from text_generation import Client
-from conversation import get_default_conv_template
+from conversation import get_conv_template
 from transformers import AutoTokenizer
 from pymongo import MongoClient
 
@@ -141,7 +141,7 @@ with gr.Blocks() as demo:
 
 
     def bot(history, max_new_tokens, temperature, top_p, top_k, system_prompt):
-        conv = get_default_conv_template("twllm_v2").copy()
+        conv = get_conv_template("twllm_v2").copy()
         roles = {"human": conv.roles[0], "gpt": conv.roles[1]}  # map human to USER and gpt to ASSISTANT
         conv.system = system_prompt
         for user, bot in history: