update(README): fix code snippet

README.md

@@ -35,14 +35,14 @@ This format is also available through the tokenizer's `apply_chat_template` meth
 ```python
 from transformers import AutoModelForCausalLM, AutoTokenizer
 
-tokenizer = AutoTokenizer.from_pretrained('stabilityai/stablelm-2-chat', trust_remote_code=True)
+tokenizer = AutoTokenizer.from_pretrained('stabilityai/stablelm-2-12b-chat')
 model = AutoModelForCausalLM.from_pretrained(
-    'stabilityai/stablelm-2-chat',
+    'stabilityai/stablelm-2-12b-chat',
     device_map="auto",
     trust_remote_code=True,
 )
 
-prompt = [{'role': 'user', 'content': 'How to achieve multiple rows of data into one row of data in Excel?'}]
+prompt = [{'role': 'user', 'content': 'How to combine multiple rows of data into one row of data in Excel?'}]
 inputs = tokenizer.apply_chat_template(
     prompt,
     add_generation_prompt=True,
@@ -55,14 +55,14 @@ tokens = model.generate(
     temperature=0.7,
     do_sample=True
 )
-output = tokenizer.decode(tokens[:, inputs.input_ids.shape[-1]:][0], skip_special_tokens=False)
+output = tokenizer.decode(tokens[:, inputs.shape[-1]:][0], skip_special_tokens=False)
 
 print(output)
 ```
 
 StableLM 2 12B Chat also supports function call usage this is an example how you can use it:
 ```python
-system_prompt_func = """\
+system_prompt = """\
 You are a helpful assistant with access to the following functions. You must use them if required -\n
 [
   {
@@ -103,7 +103,7 @@ tokens = model.generate(
     temperature=0.5,
     do_sample=True
 )
-output = tokenizer.decode(tokens[:, inputs.input_ids.shape[-1]:][0], skip_special_tokens=False)
+output = tokenizer.decode(tokens[:, inputs.shape[-1]:][0], skip_special_tokens=True)
 
 print(output)
 """