Update README.md

README.md

@@ -24,8 +24,8 @@ We cast this task as a multi-class classification task, where the objective is g
 from transformers import AutoTokenizer, AutoModelForSequenceClassification
 import torch
 
-tokenizer = AutoTokenizer.from_pretrained("gabski/deberta-claim-improvement-suggestion-with-parent-context")
-model = AutoModelForSequenceClassification.from_pretrained("gabski/deberta-claim-improvement-suggestion-with-parent-context")
+tokenizer = AutoTokenizer.from_pretrained("gabski/deberta-claim-improvement-suggestion-with-thesis-context")
+model = AutoModelForSequenceClassification.from_pretrained("gabski/deberta-claim-improvement-suggestion-with-thesis-context")
 claim = 'Teachers are likely to educate children better than parents.'
 thesis = 'Homeschooling should be banned.'
 model_input = tokenizer(claim,thesis, return_tensors='pt')