enotkrutoy commited on
Commit
2bca226
·
verified ·
1 Parent(s): ce69bda

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -23,7 +23,7 @@ def respond(
23
 
24
  messages.append({"role": "user", "content": message})
25
 
26
- input_ids = tokenizer.encode(messages, return_tensors="pt")
27
 
28
  output = model.generate(
29
  input_ids,
 
23
 
24
  messages.append({"role": "user", "content": message})
25
 
26
+ input_ids = tokenizer.encode(" ".join([m["content"] for m in messages]), return_tensors="pt")
27
 
28
  output = model.generate(
29
  input_ids,