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

f92adcb

•

1 Parent(s): af297b6

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -30,7 +30,7 @@ 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)
     print(output)
     return output

 model = AutoModelForMultipleChoice.from_pretrained(model_name, config = config)
 def run_model(input_string):
+    input_ids = tokenizer.encode(input_string, return_tensors="pt")
     output = model(input_ids)
     print(output)
     return output