Update README.md

README.md

@@ -20,10 +20,10 @@ This is a reward model based on Bert Uncased.
 model_name = "entfane/BERT_human_like_RM"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForSequenceClassification.from_pretrained(model_name)
-messages = ["How are you doing? Great, wbu?",
-"How are you doing?" Greetings! I am doing just fine, may I ask you, how are you doing?
+messages = ["How are you doing? Great",
+"How are you doing? Greetings! I am doing just fine, may I ask you, how are you doing?"
 ]
 input = tokenizer(messages, return_tensors="pt", padding="max_length").to(model.device)
-output = model(**encoded_input)
+output = model(**input)
 print(output)
 ```