Update README.md

README.md

@@ -7,7 +7,7 @@ language:
 GPT-J for preference modeling
 
 - Dataset: https://huggingface.co/datasets/reciprocate/oasst_hh_shp_hellaswag_webgpt_rm_dataset
-- Logs: https://wandb.ai/sorry/autocrit/runs/y9des5kz
+- Logs: https://wandb.ai/sorry/autocrit/runs/y9des5kz?workspace=user-sorry
 
 Usage:
 
@@ -18,5 +18,10 @@ tokenizer = AutoTokenizer.from_pretrained("reciprocate/gpt-j_rm_format-oa", revi
 model = AutoModelForSequenceClassification.from_pretrained("reciprocate/gpt-j_rm_format-oa", revision="501f895")
 
 input = "<|prompter|>Are you lying right now?</s><|assistant|>I am :)</s><|endoftext|>"
-score = model(**tokenizer(input, return_tensors="pt"))[0]
+model(**tokenizer(input, return_tensors="pt"))[0].item()
 ```
+
+Output:
+```python
+-4.877448558807373
+```