yentinglin commited on
Commit
fe7697d
·
1 Parent(s): 032e12f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -28,6 +28,7 @@ def generate_response(user_input, max_new_token, top_p, top_k, temperature, do_s
28
  {"role": "user", "content": user_input},
29
  ])
30
  msg += "<|assistant|>\n"
 
31
 
32
  res = client.generate(
33
  msg,
 
28
  {"role": "user", "content": user_input},
29
  ])
30
  msg += "<|assistant|>\n"
31
+ msg = user_input.strip()
32
 
33
  res = client.generate(
34
  msg,