File size: 225 Bytes
c10350f
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
# coding=utf-8
# author: xusong <xusong28@jd.com>
# time: 2022/8/25 15:49


from transformers import pipeline

if __name__ == "__main__":
    classifier = pipeline("fill-mask")
    classifier("Paris is the <mask> of France.")