Update app.py

app.py

@@ -5,7 +5,9 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
 import spaces
   
 tokenizer = AutoTokenizer.from_pretrained("Rorical/0-roleplay", trust_remote_code=True)
-tokenizer.chat_template = "{% for message in messages %}{{'<|im_start|>' + ((message['role'] + ':\n') if message['role'] != '' else '') + message['content'] + '<|im_end|>' + '\n'}}{% endfor %}{% if add_generation_prompt %}{{ '<|im_start|>星野:\n' }}{% endif %}" # Be careful that this model used custom chat template.
+tokenizer.chat_template = "{% for message in messages %}{{'<|im_start|>' + ((message['role'] + '\n') if message['role'] != '' else '') + message['content'] + '<|im_end|>' + '\n'}}{% endfor %}"
+tokenizer.add_special_tokens({"bos_token": tokenizer.eos_token})
+tokenizer.bos_token_id = tokenizer.eos_token_id
 
 # Define the response function
 @spaces.GPU