Update README.md

README.md

@@ -79,7 +79,7 @@ tokens = tokenizer(question, response1,
 tokens.to(device)
 
 # Score the response
-score = model(**tokens, alpha=10).item()
+score = rewardModel(**tokens, alpha=10).item()
 
 print(f"Question: {question} \n")
 print(f"Response 1: {response1} Score: {score:.3f}")
@@ -91,7 +91,7 @@ tokens = tokenizer(question, response2,
 
 tokens.to(device)
 
-score = model(**tokens, alpha=10).item()
+score = rewardModel(**tokens, alpha=10).item()
 
 print(f"Response 2: {response2} Score: {score:.3f}")
 ```