Update README.md

README.md

@@ -16,7 +16,7 @@ Paper: [To Revise or Not to Revise: Learning to Detect Improvable Claims for Arg
 Authors: Gabriella Skitalinskaya and Henning Wachsmuth
 
 # Claim Improvement Suggestion
-We cast this task as a multi-class classification task, where the objective is given an argumentative claim and some contextual information (in this case, the parent claim in the debate, which is opposed or supported by the claim in question), select all types of quality issues from a defined set that should be improved when revising the claim.
+We cast this task as a multi-class classification task, where the objective is given an argumentative claim and some contextual information (in this case, the **parent claim** in the debate, which is opposed or supported by the claim in question), select all types of quality issues from a defined set that should be improved when revising the claim.
 
 # Usage
 
@@ -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("deberta-claim-improvement-suggestion-with-parent-context")
-model = AutoModelForSequenceClassification.from_pretrained("deberta-claim-improvement-suggestion-with-parent-context")
+tokenizer = AutoTokenizer.from_pretrained("gabski/deberta-claim-improvement-suggestion-with-parent-context")
+model = AutoModelForSequenceClassification.from_pretrained("gabski/deberta-claim-improvement-suggestion-with-parent-context")
 claim = 'Teachers are likely to educate children better than parents.'
 parent_claim = 'Homeschooling should be banned.'
 model_input = tokenizer(claim,parent_claim, return_tensors='pt')