Leet-Search / backend /app /utils /get_embeddings.py
Eli4479
Added FastAPI backend for Leet Search
c8fa3bd
raw
history blame contribute delete
262 Bytes
from sentence_transformers import SentenceTransformer
model = SentenceTransformer("BAAI/bge-base-en")
def get_embedding(text: str):
prompt = f"Represent this sentence for search: {text}"
return model.encode(prompt, normalize_embeddings=True).tolist()