File size: 494 Bytes
c10350f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 |
# 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) |