persiannlp
/

wikibert-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

ef4004e

•

1 Parent(s): d035e66

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -29,7 +29,7 @@ model = BertModel.from_pretrained(model_name)
 def run_model(input_string):
-    input_ids = tokenizer.encode(input_string, return_tensors="pt")
     output = model(input_ids)
     print(output)
     return output

 def run_model(input_string):
+    input_ids = tokenizer(input_string, return_tensors="pt")
     output = model(input_ids)
     print(output)
     return output