src/embedding.py

from sentence_transformers import SentenceTransformer
import streamlit as st

@st.cache_resource
def get_embed_model():
    return SentenceTransformer("all-MiniLM-L6-v2")

def get_embeddings(texts):
    model = get_embed_model()
    return model.encode(texts, convert_to_numpy=True)