Update README.md

README.md

@@ -22,11 +22,12 @@ This is a wikibert-based model for multiple-choice question answering.
 Here is an example of how you can run this model: 
 
 ```python 
-from transformers import BertTokenizer, BertModel
+from transformers import AutoConfig, AutoModelForMultipleChoice, AutoTokenizer
 
 model_name = "persiannlp/wikibert-base-parsinlu-multiple-choice"
-tokenizer = BertTokenizer.from_pretrained(model_name)
-model = BertModel.from_pretrained(model_name)
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+config = AutoConfig.from_pretrained(model_name)
+model = AutoModelForMultipleChoice.from_pretrained(model_name, config = config)
 
 
 def run_model(input_string):