Update code example (#11631)
Update code example (#11631)

* Update code example

* Code review
huggingface/transformers,huggingface/transformers,huggingface/pytorch-transformers,huggingface/transformers,huggingface/transformers