Update README.md

README.md

@@ -456,7 +456,9 @@ output = model.generate(input_ids,
                         no_repeat_ngram_size=3,
                         num_return_sequences=1,
                         )
-print(tokenizer.decode(output[0], skip_special_tokens=False))
+input_length = input_ids.shape[1]
+generated_tokens = output[0][input_length:]
+print(tokenizer.decode(generated_tokens, skip_special_tokens=True))
 ```
 ## Disclaimer