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)))