Update README.md

README.md

@@ -32,8 +32,15 @@ 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")
+bert_base = "bert-base-chinese"
+model_id = "Xiaoxi2333/bert_multilabel_chinese"
+tokenizer = BertTokenizer.from_pretrained(model_id)
+model = BertMultiLabelClassifier(pretrained_model_path=bert_base, num_labels=5)
+state_dict = torch.hub.load_state_dict_from_url(
+    f"https://huggingface.co/{model_id}/resolve/main/pytorch_model.bin",
+    map_location="cpu"
+)
+model.load_state_dict(state_dict)
 model.eval()
 
 # 定义标签