pipeline/embedding.py

from sentence_transformers import SentenceTransformer

class Embedder:
    def __init__(self, model_name='all-MiniLM-L6-v2'):
        self.model = SentenceTransformer(model_name)

    def encode(self, texts, batch_size=32, show_progress_bar=False):
        return self.model.encode(texts, batch_size=batch_size, show_progress_bar=show_progress_bar)