Felladrin commited on
Commit
dca8bdb
1 Parent(s): e056027

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -20,8 +20,8 @@ def generate(
20
  "role": "system",
21
  "content": system_input,
22
  },
23
- {"role": "user", "content": "Hello there!"},
24
- {"role": "assistant", "content": "Hi!"},
25
  {"role": "user", "content": user_input},
26
  ]
27
  prompt = pipe.tokenizer.apply_chat_template(message_template, tokenize=False, add_generation_prompt=True)
 
20
  "role": "system",
21
  "content": system_input,
22
  },
23
+ {"role": "user", "content": "Hi!"},
24
+ {"role": "assistant", "content": "Hello there. How can I assist you today?"},
25
  {"role": "user", "content": user_input},
26
  ]
27
  prompt = pipe.tokenizer.apply_chat_template(message_template, tokenize=False, add_generation_prompt=True)