nroggendorff commited on
Commit
e6142fd
1 Parent(s): 7e0dfdf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ def generate_chatml(messages):
5
  chatml = ""
6
  for i, message in enumerate(messages):
7
  if message.strip():
8
- role = "<|im_start|>user\n" if i % 2 == 0 else "<|im_start|>assistant\n"
9
  chatml += f"{role}\n{message.strip()}\n<|im_end|>\n"
10
  return chatml.strip()
11
 
 
5
  chatml = ""
6
  for i, message in enumerate(messages):
7
  if message.strip():
8
+ role = "<|im_start|>user" if i % 2 == 0 else "<|im_start|>assistant"
9
  chatml += f"{role}\n{message.strip()}\n<|im_end|>\n"
10
  return chatml.strip()
11