NYCourtOrderClass / utils.py
szlevi's picture
Upload folder using huggingface_hub
813dc3a verified
import chromadb
from sentence_transformers import SentenceTransformer
chromaclient = chromadb.Client()
collection = chromaclient.get_or_create_collection("court_orders")
embedding_model = SentenceTransformer("all-MiniLM-L6-v2")
def gettextembed(text):
return embedding_model.encode([text])[0]