Update README.md

README.md

@@ -28,10 +28,11 @@ tokenizer = BertTokenizer.from_pretrained(model_name)
 model = BertModel.from_pretrained(model_name)
 
 
-def run_model(input_string, **generator_args):
+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)
+    #res = model.generate(input_ids, **generator_args)
+    #output = tokenizer.batch_decode(res, skip_special_tokens=True)
+    output = model(input_ids)
     print(output)
     return output