Partha11 commited on
Commit
61739eb
·
verified ·
1 Parent(s): bcf8a29

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -2
app.py CHANGED
@@ -26,8 +26,14 @@ def respond(
26
  top_p,
27
  ):
28
  messages = [
29
- system_message,
30
- message
 
 
 
 
 
 
31
  ]
32
  prompt = tokenizer.apply_chat_template(
33
  messages,
 
26
  top_p,
27
  ):
28
  messages = [
29
+ {
30
+ "role": "system",
31
+ "content": system_message
32
+ },
33
+ {
34
+ "role": "user",
35
+ "content": message
36
+ }
37
  ]
38
  prompt = tokenizer.apply_chat_template(
39
  messages,