persiannlp
/

mbert-base-parsinlu-multiple-choice

Text Classification

multiple-choice

Inference Endpoints

Model card Files Files and versions Community

pouya-pezeshkpour commited on Feb 28, 2021

Commit

b61207b

•

1 Parent(s): 47adf64

Update README.md

Files changed (1) hide show

README.md +4 -3

README.md CHANGED Viewed

@@ -28,10 +28,11 @@ tokenizer = BertTokenizer.from_pretrained(model_name)
 model = BertModel.from_pretrained(model_name)
-def run_model(input_string, **generator_args):
     input_ids = tokenizer.encode(input_string, return_tensors="pt")
-    res = model.generate(input_ids, **generator_args)
-    output = tokenizer.batch_decode(res, skip_special_tokens=True)
     print(output)
     return output

 model = BertModel.from_pretrained(model_name)
+def run_model(input_string):
     input_ids = tokenizer.encode(input_string, return_tensors="pt")
+    #res = model.generate(input_ids, **generator_args)
+    #output = tokenizer.batch_decode(res, skip_special_tokens=True)
+    output = model(input_ids)
     print(output)
     return output