Update README.md

README.md

@@ -123,18 +123,13 @@ generated_ids = model.generate(
 )
 output_ids = generated_ids[0][len(model_inputs.input_ids[0]):].tolist() 
 
-# parsing thinking content
-try:
-    # rindex finding 151668 (</think>)
-    index = len(output_ids) - output_ids[::-1].index(151668)
-except ValueError:
-    index = 0
-
-thinking_content = tokenizer.decode(output_ids[:index], skip_special_tokens=True).strip("\n")
-content = tokenizer.decode(output_ids[index:], skip_special_tokens=True).strip("\n")
-
-print("thinking content:", thinking_content) # no opening <think> tag
-print("content:", content)
+# parsing output - if you get errors, make sure you update to the latest version of transformers!
+response = tokenizer.parse_response(tokenizer.decode(output_ids))
+
+# If any tool calls are present, they will be in response.tool_calls
+# You can continue the chat by simply appending `response` to `messages`.
+print("thinking content:", response.reasoning)
+print("content:", response.content)
 
 ```