krishna0506/Document_Intelligence

1from sentence_transformers import SentenceTransformer
2
3model = SentenceTransformer("all-MiniLM-L6-v2")
4
5def embed_texts(texts):
6    return model.encode(texts, normalize_embeddings=True)
7
8def embed_query(query):
9    return model.encode([query], normalize_embeddings=True)