Update example

README.md

@@ -93,8 +93,10 @@ prompt = tokenizer.apply_chat_template(
 
 inputs = tokenizer.encode(prompt, add_special_tokens=False, return_tensors="pt")
 outputs = model.generate(input_ids=inputs.to(model.device), max_new_tokens=200)
-
-print(tokenizer.decode(outputs[0], skip_special_tokens=True))
+generated_tokens = outputs[0][len(inputs[0]):]
+response = self.tokenizer.decode(generated_tokens, skip_special_tokens=False).strip()
+response = response.split("<|reserved_token_1|>")[0].strip()
+print(response)
 
 ```