PraneetNS commited on
Commit
7ae37cd
·
verified ·
1 Parent(s): aa88a35

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -0
app.py CHANGED
@@ -87,6 +87,12 @@ def generate(message, history):
87
  "content": content,
88
  }
89
  )
 
 
 
 
 
 
90
 
91
  inputs = tokenizer.apply_chat_template(
92
  messages,
 
87
  "content": content,
88
  }
89
  )
90
+ messages.append(
91
+ {
92
+ "role": "user",
93
+ "content": message,
94
+ }
95
+ )
96
 
97
  inputs = tokenizer.apply_chat_template(
98
  messages,