Update README.md

README.md

@@ -20,7 +20,7 @@ from transformers import RobertaTokenizer, RobertaForSequenceClassification
 model = RobertaForSequenceClassification.from_pretrained('alisawuffles/roberta-large-wanli')
 tokenizer = RobertaTokenizer.from_pretrained('alisawuffles/roberta-large-wanli')
 
-x = tokenizer("I almost forgot to eat lunch.", "I didn't forget to eat lunch.", hypothesis, return_tensors='pt', max_length=128, truncation=True)
+x = tokenizer("I almost forgot to eat lunch.", "I didn't forget to eat lunch.", return_tensors='pt', max_length=128, truncation=True)
 logits = model(**x).logits
 probs = logits.softmax(dim=1).squeeze(0)
 label_id = torch.argmax(probs).item()