Update use_model.py

use_model.py

@@ -8,11 +8,10 @@ from tokenization import WoBertTokenizer
 from transformers.models.bert import BertModel, BertConfig
 
 input_sentent = '欢迎使用阿里灵界比赛预训练模型'
-tokenizer = WoBertTokenizer.from_pretrained('./wobert_repretrain/vocab.txt')
+tokenizer = WoBertTokenizer.from_pretrained('luxiao/alilingjie')
 input_ids = tokenizer.encode(input_sentent, return_tensors='pt')
 
-config = BertConfig.from_pretrained('./wobert_repretrain/config.json')
-model = BertModel.from_pretrained('./wobert_repretrain/pytorch_model.bin', config=config)
+model = BertModel.from_pretrained('luxiao/alilingjie')
 
 output = model(input_ids)
 print(output[0].size())   # torch.Size([1, 11, 768])