|
from sentence_transformers import SentenceTransformer
|
|
from app.config import settings
|
|
|
|
class Enricher:
|
|
def __init__(self):
|
|
self.model = SentenceTransformer(settings.EMBED_MODEL)
|
|
def embed(self, text: str):
|
|
return self.model.encode(text, normalize_embeddings=True).tolist()
|