File size: 229 Bytes
bc7dd2a | 1 2 3 4 5 6 | from modeling_kenga_embed import KengaEmbed
m = KengaEmbed.from_pretrained(".")
q = m.encode_queries(["Where is the capital of Russia?"])
d = m.encode_documents(["Moscow is the capital of Russia."])
print(float((q * d).sum(-1)))
|