Update README.md

README.md

@@ -50,7 +50,7 @@ Here is how to use this model to get the features of a given text in PyTorch:
 ```python
 from transformers import RobertaTokenizer, RobertaModel
 tokenizer = RobertaTokenizer.from_pretrained('fav-kky/FERNET-C5-RoBERTa')
-model = RobertaModel.from_pretrained('fav-kky/FERNET-C5-RoBERTa')
+model = RobertaModel.from_pretrained('fav-kky/FERNET-C5-RoBERTa', add_pooling_layer=False)
 text = "Libovolný text."
 encoded_input = tokenizer(text, return_tensors='pt')
 output = model(**encoded_input)