minor code clarification

README.md

@@ -13,14 +13,16 @@ Compute was generously provided by [Stability AI](https://stability.ai/)
 ### How to use
 
 ```python
+from transformers import AutoModelForSequenceClassification, AutoTokenizer
 # install open assistant model_training module (e.g. run `pip install -e .` in `model/` directory of open-assistant repository)
 import model_training.models.reward_model  # noqa: F401 (registers reward model for AutoModel loading)
 
+model_name = "OpenAssistant/oasst-rm-2-pythia-6.9b-epoch-1"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForSequenceClassification.from_pretrained(model_name)
 input_text = "<|prompter|>Hi how are you?<|endoftext|><|assistant|>Hi, I am Open-Assistant a large open-source language model trained by LAION AI. How can I help you today?<|endoftext|>"
 inputs = tokenizer(input_text, return_tensors="pt")
-score = rm(**inputs).logits[0].cpu().detach()
+score = model(**inputs).logits[0].cpu().detach()
 print(score)
 ```