File size: 247 Bytes
456ae8a | 1 2 3 4 5 6 7 8 9 | from mt_model import BERTcForMT, PieceCharTokenizer
tok = PieceCharTokenizer(".")
model = BERTcForMT.from_pretrained(".")
result = model.predict("中华人民共和国万岁", tok)
print(result["words"])
print(result["pos"])
print(result["ner"])
|