yentinglin commited on
Commit
628629b
1 Parent(s): 0cfd320

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -29,6 +29,7 @@ def generate_response(user_input, max_new_token, top_p, top_k, temperature, do_s
29
  roles = {"human": conv.roles[0], "gpt": conv.roles[1]} # map human to USER and gpt to ASSISTANT
30
  role = roles["human"]
31
  conv.append_message(role, user_input)
 
32
  msg = conv.get_prompt()
33
 
34
  res = client.generate(
 
29
  roles = {"human": conv.roles[0], "gpt": conv.roles[1]} # map human to USER and gpt to ASSISTANT
30
  role = roles["human"]
31
  conv.append_message(role, user_input)
32
+ conv.append_message(roles["gpt"], None)
33
  msg = conv.get_prompt()
34
 
35
  res = client.generate(