cszhzleo commited on
Commit
7536556
1 Parent(s): 3854982

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -91,7 +91,7 @@ def format_prompt(message, history):
91
  '''
92
 
93
  messages = [
94
- {"role": "user", "content": "Can you tell me an interesting fact about AWS?"},]
95
  prompt = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
96
 
97
  return prompt
 
91
  '''
92
 
93
  messages = [
94
+ {"role": "user", "content": message},]
95
  prompt = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
96
 
97
  return prompt