Update README.md

README.md

@@ -66,12 +66,12 @@ pip install transformers torch
 
 ### Load Model and Tokenizer
 ```
-from transformers import DistilBertForSequenceClassification, DistilBertTokenizer
+from transformers import AutoTokenizer, AutoModelForSequenceClassification
 
 # Load model and tokenizer
 model_name = "distilbert-base-q-cat"
-model = DistilBertForSequenceClassification.from_pretrained(model_name)
-tokenizer = DistilBertTokenizer.from_pretrained(model_name)
+model = AutoModelForSequenceClassification.from_pretrained(model_name, num_labels=3, ignore_mismatched_sizes=True)
+tokenizer = AutoTokenizer.from_pretrained(model_name)
 ```
 
 ### Inference Example