wanyu commited on
Commit
d9e6e6e
1 Parent(s): 9792969

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -67,8 +67,8 @@ More specifically, the model will predict the probability of the following edit
67
  import torch
68
  from transformers import AutoTokenizer, AutoModelForSequenceClassification
69
 
70
- tokenizer = AutoTokenizer.from_pretrained("wanyu/IteraTeR-ROBERTA")
71
- model = AutoModelForSequenceClassification.from_pretrained("wanyu/IteraTeR-ROBERTA")
72
 
73
  id2label = {0: "clarity", 1: "fluency", 2: "coherence", 3: "style", 4: "meaning-changed"}
74
 
 
67
  import torch
68
  from transformers import AutoTokenizer, AutoModelForSequenceClassification
69
 
70
+ tokenizer = AutoTokenizer.from_pretrained("wanyu/IteraTeR-ROBERTA-Intention-Classifier")
71
+ model = AutoModelForSequenceClassification.from_pretrained("wanyu/IteraTeR-ROBERTA-Intention-Classifier")
72
 
73
  id2label = {0: "clarity", 1: "fluency", 2: "coherence", 3: "style", 4: "meaning-changed"}
74