tanyuzhou commited on
Commit
d1a3973
1 Parent(s): 4633b4e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -46,7 +46,7 @@ def respond(
46
 
47
  # Decode the generated response
48
  response = tokenizer.decode(generate_ids[0], skip_special_tokens=True)
49
- response = response.split(f"老师\n{message}\n星野\n")[1]
50
 
51
  return response
52
 
 
46
 
47
  # Decode the generated response
48
  response = tokenizer.decode(generate_ids[0], skip_special_tokens=True)
49
+ response = response.split(f"{user_name}\n{message}\n{bot_name}\n")[1]
50
 
51
  return response
52