wanyu
/

IteraTeR-ROBERTA-Intention-Classifier

Text Classification

Model card Files Files and versions

wanyu commited on Mar 20, 2022

Commit

d9e6e6e

·

1 Parent(s): 9792969

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -67,8 +67,8 @@ More specifically, the model will predict the probability of the following edit
 import torch
 from transformers import AutoTokenizer, AutoModelForSequenceClassification
-tokenizer = AutoTokenizer.from_pretrained("wanyu/IteraTeR-ROBERTA")
-model = AutoModelForSequenceClassification.from_pretrained("wanyu/IteraTeR-ROBERTA")
 id2label = {0: "clarity", 1: "fluency", 2: "coherence", 3: "style", 4: "meaning-changed"}

 import torch
 from transformers import AutoTokenizer, AutoModelForSequenceClassification
+tokenizer = AutoTokenizer.from_pretrained("wanyu/IteraTeR-ROBERTA-Intention-Classifier")
+model = AutoModelForSequenceClassification.from_pretrained("wanyu/IteraTeR-ROBERTA-Intention-Classifier")
 id2label = {0: "clarity", 1: "fluency", 2: "coherence", 3: "style", 4: "meaning-changed"}