mvi-ai-engine / memory /embeder.py
Musombi's picture
Upload folder using huggingface_hub
0045f6d verified
from sentence_transformers import SentenceTransformer
class Embedder:
def __init__(self, model="sentence-transformers/all-MiniLM-L6-v2"):
self.model = SentenceTransformer(model)
def encode(self, text):
return self.model.encode(text).tolist()