Update README.md

README.md

@@ -4,13 +4,10 @@ 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
+from transformers import AutoTokenizer, GPT2ForSequenceClassificat
+tokenizer = AutoTokenizer.from_pretrained("microsoft/DialogRPT-updown")
+model = GPT2ForSequenceClassification.from_pretrained("sugam11/gpt2-rlhf-rewar
+inputs = tokenizer("Hello, my dog is cute", return_tensors="p
+with torch.no_grad():
+  logits = model(**inputs).logits