embeddings.py

from sentence_transformers import SentenceTransformer

# Load the pre-trained SentenceTransformer model
model = SentenceTransformer("sentence-transformers/all-MiniLM-L6-v2")

def get_embedding(text):
    """Generate embeddings for a given text using SentenceTransformer."""
    return model.encode([text])[0]  # Return the first embedding (only one sentence)