Spaces:

NitinMoturu
/

maiseumsChat

Sleeping

maiseumsChat / embedding_utils.py

Create embedding_utils.py

21f3194 verified about 1 year ago

168 Bytes

from sentence_transformers import SentenceTransformer

model = SentenceTransformer("all-MiniLM-L6-v2")

def get_embedding(text):
    return model.encode(text).tolist()

	from sentence_transformers import SentenceTransformer

	model = SentenceTransformer("all-MiniLM-L6-v2")

	def get_embedding(text):
	return model.encode(text).tolist()