sentence-transformers How to use Nutanix/jina-embeddings-v2-base-code-mbpp with sentence-transformers:
from sentence_transformers import SentenceTransformer
model = SentenceTransformer("Nutanix/jina-embeddings-v2-base-code-mbpp", trust_remote_code=True)
sentences = [
"Write a function to extract every specified element from a given two dimensional list.",
"def nCr_mod_p(n, r, p): \r\n\tif (r > n- r): \r\n\t\tr = n - r \r\n\tC = [0 for i in range(r + 1)] \r\n\tC[0] = 1 \r\n\tfor i in range(1, n + 1): \r\n\t\tfor j in range(min(i, r), 0, -1): \r\n\t\t\tC[j] = (C[j] + C[j-1]) % p \r\n\treturn C[r] ",
"import cmath\r\ndef len_complex(a,b):\r\n cn=complex(a,b)\r\n length=abs(cn)\r\n return length",
"def specified_element(nums, N):\r\n result = [i[N] for i in nums]\r\n return result"
]
embeddings = model.encode(sentences)
similarities = model.similarity(embeddings, embeddings)
print(similarities.shape)
# [4, 4] 
