Update README.md

README.md

@@ -42,10 +42,10 @@ Fine-tuned version of the original huBERT model (`SZTAKI-HLT/hubert-base-cc`), t
 ## Usage
 
 ```py
-from transformers import AutoTokenizer, AutoModel
+from transformers import AutoTokenizer, AutoModelForSequenceClassification
 
 tokenizer = AutoTokenizer.from_pretrained("uvegesistvan/huBERTPlain")
-model = AutoModel.from_pretrained("uvegesistvan/huBERTPlain")
+model = AutoModelForSequenceClassification.from_pretrained("uvegesistvan/huBERTPlain")
 ```
 
 ### BibTeX entry and citation info