Update README.md

README.md

@@ -2,4 +2,15 @@
 license: mit
 datasets:
 - Anthropic/hh-rlhf
----
+---
+```python
+from transformers import AutoTokenizer, GPT2ForSequenceClassification
+
+tokenizer = AutoTokenizer.from_pretrained("microsoft/DialogRPT-updown")
+model = GPT2ForSequenceClassification.from_pretrained("sugam11/gpt2-rlhf-reward")
+
+inputs = tokenizer("Hello, my dog is cute", return_tensors="pt")
+
+with torch.no_grad():
+    logits = model(**inputs).logits
+