Spaces:
Sleeping
Sleeping
File size: 509 Bytes
969189d 473c307 22b9832 473c307 22b9832 969189d 473c307 969189d | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 | from sentence_transformers import SentenceTransformer
# from app.config import EMBED_MODEL_NAME
import os
os.environ['TRANSFORMERS_CACHE'] = '/tmp/transformers_cache'
os.environ['HF_HOME'] = '/tmp/huggingface'
os.environ['SENTENCE_TRANSFORMERS_HOME'] = '/tmp/sentence_transformers'
embed_model = SentenceTransformer("all-MiniLM-L6-v2")
def embed_texts(texts):
return embed_model.encode(texts, convert_to_numpy=True)
def embed_text(text):
return embed_model.encode([text], convert_to_numpy=True)[0]
|