rwitz commited on
Commit
c621c14
·
1 Parent(s): 05f18ad

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -28,9 +28,9 @@ def write_elo_ratings(elo_ratings):
28
 
29
  # Function to get bot response
30
  def format_chatml_prompt(state):
31
- chatml_prompt = "<s>"
32
  for message in state["history"]:
33
- chatml_prompt += message['role']+": "+ message['content']+"\n"
34
  return chatml_prompt+"Assistant:"
35
  def get_bot_response(url, prompt,state):
36
  chatml_prompt = format_chatml_prompt(state)
 
28
 
29
  # Function to get bot response
30
  def format_chatml_prompt(state):
31
+ chatml_prompt = ""
32
  for message in state["history"]:
33
+ chatml_prompt += "<s>"+message['role']+": "+ message['content']+"</s>\n"
34
  return chatml_prompt+"Assistant:"
35
  def get_bot_response(url, prompt,state):
36
  chatml_prompt = format_chatml_prompt(state)