Xiaoxi2333
/

bert_multilabel_chinese

multilabel-classification

intent-classification

Model card Files Files and versions

Xiaoxi2333 commited on Jun 12

Commit

4a9b420

·

verified ·

1 Parent(s): 1760b74

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -32,8 +32,8 @@ from transformers import BertTokenizer
 from bert_classifier_3 import BertMultiLabelClassifier
 # 加载 tokenizer 和模型
-tokenizer = BertTokenizer.from_pretrained("Xiaoxi2333/bert-multilabel-chinese")
-model = BertMultiLabelClassifier(pretrained_model_path="Xiaoxi2333/bert-multilabel-chinese")
 model.load_state_dict(torch.load("pytorch_model.bin", map_location="cpu"))
 model.eval()

 from bert_classifier_3 import BertMultiLabelClassifier
 # 加载 tokenizer 和模型
+tokenizer = BertTokenizer.from_pretrained("Xiaoxi2333/bert_multilabel_chinese")
+model = BertMultiLabelClassifier(pretrained_model_path="Xiaoxi2333/bert_multilabel_chinese")
 model.load_state_dict(torch.load("pytorch_model.bin", map_location="cpu"))
 model.eval()