persiannlp
/

mbert-base-parsinlu-multiple-choice

Text Classification

multiple-choice

Model card Files Files and versions

pouya-pezeshkpour commited on Feb 28, 2021

Commit

20a09df

·

1 Parent(s): 8549907

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -25,7 +25,7 @@ from transformers import BertTokenizer, BertModel
 model_name = f"persiannlp/mt5-base-parsinlu-multiple-choice"
 tokenizer = BertTokenizer.from_pretrained(model_name)
-model = BertModelForConditionalGeneration.from_pretrained(model_name)
 def run_model(input_string, **generator_args):

 model_name = f"persiannlp/mt5-base-parsinlu-multiple-choice"
 tokenizer = BertTokenizer.from_pretrained(model_name)
+model = BertModel.from_pretrained(model_name)
 def run_model(input_string, **generator_args):