# coding=utf-8 | |
# author: xusong <xusong28@jd.com> | |
# time: 2022/8/25 10:19 | |
""" | |
未完成 | |
https://github.com/shibing624/pycorrector/blob/6f1b6ea35c60700f08463950e11ae0963ed512b2/pycorrector/bert/bert_corrector.py#L30 | |
""" | |
import register | |
from transformers import pipeline | |
bert_model_dir = "models/pretrain/" | |
model = pipeline('fill-mask', | |
model=bert_model_dir, | |
tokenizer=bert_model_dir) | |
predicts = model("这款连[MASK]裙真漂亮") | |
print(predicts) |