# Vergleiche verschiedene SBERT 
print("Wir laden")
from sentence_transformers import SentenceTransformer, util

muster="Katze frisst Maus"
vergl=["Katze verspeist Maus","Katze liebt Maus","Katze verzehrt Maus","Katze verehrt Maus"]
modnam=["T-Systems-onsite/german-roberta-sentence-transformer-v2","deepset/gbert-base","aari1995/German_Semantic_STS_V2"]

print ("Vergleich mit ",muster)
for mona in modnam:
    print("\n",mona)
    model = SentenceTransformer(mona) 
    for v in vergl:
        emb1, emb2 = model.encode([muster, v], convert_to_tensor=True)
        print("%-20s   %5.3f"%(v, util.cos_sim(emb1, emb2).item()))