Spaces:
Runtime error
Runtime error
File size: 262 Bytes
c8fa3bd |
1 2 3 4 5 6 7 8 9 |
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()
|