from nlpcda import Simbert

simbert_config = {
    'model_path': './classify_service/chinese_roformer-sim-char_L-12_H-768_A-12',
    'CUDA_VISIBLE_DEVICES': '0,1',
    'max_len': 64,
    'seed': 1
}
if __name__ == '__main__':
    l = ["登录", "注册", "注销", "点击", "关闭"]
    for item in l:
        simbert = Simbert(config=simbert_config)
        synonyms = simbert.replace(sent=l, create_num=5)
        print(item, synonyms)