nikravan commited on
Commit
2ed7265
Β·
verified Β·
1 Parent(s): 00ab235

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -58,8 +58,8 @@ def predict(message, history, system_prompt, temperature, max_new_tokens, top_k,
58
  stop_tokens = [tokenizer.eos_token_id]
59
  instruction = system_prompt + "\n\n"
60
  for user, assistant in history:
61
- instruction += f"role:user, content: {user}\nrole:assistant, content: {assistant}\n"
62
- instruction += f"role:user, content: {message}\nassistant:"
63
 
64
  print(instruction)
65
 
 
58
  stop_tokens = [tokenizer.eos_token_id]
59
  instruction = system_prompt + "\n\n"
60
  for user, assistant in history:
61
+ instruction += f"<|User|>{user}<|Assistant|>{assistant}\n"
62
+ instruction += f"<|User|>{message}<|Assistant|>"
63
 
64
  print(instruction)
65