Update README.md

README.md

@@ -22,16 +22,16 @@ This is a mbert-based model for multiple-choice question answering.
 Here is an example of how you can run this model: 
 
 ```python 
-from transformers import BertTokenizer, BertModel
-
-model_name = "persiannlp/mbert-base-parsinlu-multiple-choice"
-tokenizer = BertTokenizer.from_pretrained(model_name)
-model = BertModel.from_pretrained(model_name)
+from transformers import AutoConfig, AutoModelForMultipleChoice, AutoTokenizer
 
+model_name = "mbert-base-parsinlu-multiple-choice"
+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):
     input_ids = tokenizer(input_string, return_tensors="pt")
-    output = model(input_ids)
+    output = model(input_ids['input_ids'])
     print(output)
     return output