synonyms.py

import pandas as pd
from sentence_transformers import SentenceTransformer
cmp_model = SentenceTransformer('uer/sbert-base-chinese-nli')

def compare(text0, text1):
    embeddings = cmp_model.encode([text0, text1])
    return pd.DataFrame(embeddings.T).corr().iloc[0, 1]