Omnibus commited on
Commit
d786de6
1 Parent(s): d7942b7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -14,12 +14,12 @@ InferenceClient(models[2]),
14
  InferenceClient(models[3]),
15
  ]
16
  def format_prompt(message, history):
17
- prompt = "<s>"
18
  if history:
19
  #<start_of_turn>userHow does the brain work?<end_of_turn><start_of_turn>model
20
  for user_prompt, bot_response in history:
21
  prompt += f"<start_of_turn>user{user_prompt}<end_of_turn>"
22
- prompt += f"<start_of_turn>model{bot_response}</s>"
23
  prompt += f"<start_of_turn>user{message}<end_of_turn><start_of_turn>model"
24
  return prompt
25
 
 
14
  InferenceClient(models[3]),
15
  ]
16
  def format_prompt(message, history):
17
+ prompt = ""
18
  if history:
19
  #<start_of_turn>userHow does the brain work?<end_of_turn><start_of_turn>model
20
  for user_prompt, bot_response in history:
21
  prompt += f"<start_of_turn>user{user_prompt}<end_of_turn>"
22
+ prompt += f"<start_of_turn>model{bot_response}"
23
  prompt += f"<start_of_turn>user{message}<end_of_turn><start_of_turn>model"
24
  return prompt
25