thu-coai
/

roberta-base-cdconv

Text Classification

Model card Files Files and versions

chujiezheng commited on Nov 2, 2022

Commit

d805257

·

1 Parent(s): 1f2af6c

Update README.md

Files changed (1) hide show

README.md +2 -3

README.md CHANGED Viewed

@@ -2,10 +2,9 @@
 ```python
 import torch
-from transformers import AutoTokenizer
-from transformers.models.bert.modeling_bert import BertForSequenceClassification
-tokenizer = AutoTokenizer.from_pretrained('chujiezheng/roberta-base-cdconv')
 model = BertForSequenceClassification.from_pretrained('chujiezheng/roberta-base-cdconv')
 model.eval()

 ```python
 import torch
+from transformers.models.bert import BertTokenizer, BertForSequenceClassification
+tokenizer = BertTokenizer.from_pretrained('chujiezheng/roberta-base-cdconv')
 model = BertForSequenceClassification.from_pretrained('chujiezheng/roberta-base-cdconv')
 model.eval()