YC-Chen commited on
Commit
aac2baa
1 Parent(s): 98ba996

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -106,7 +106,7 @@ with gr.Blocks() as demo:
106
  {"role": "user", "content": user_msg},
107
  {"role": "assistant", "content": assistant_msg},
108
  ]
109
- message = tokenizer.apply_chat_template(chat, tokenize=False)
110
  message = message[3:] # remove SOT token
111
 
112
  print(message)
 
106
  {"role": "user", "content": user_msg},
107
  {"role": "assistant", "content": assistant_msg},
108
  ]
109
+ message = tokenizer.apply_chat_template(chat_data, tokenize=False)
110
  message = message[3:] # remove SOT token
111
 
112
  print(message)