Summarize-Agent / utils /embeddings.py
nelsondiasandre's picture
Upload 13 files
9b45841 verified
Raw
History Blame Contribute Delete
228 Bytes
from sentence_transformers import SentenceTransformer
model = SentenceTransformer(
"intfloat/multilingual-e5-small"
)
def create_embeddings(chunks):
embeddings = model.encode(chunks)
return embeddings