BookingCare/multilingual-e5-base-similarity-v1

Para usar este modelo, primero instala la biblioteca Sentence Transformers:

pip install -U sentence-transformers

Luego puedes cargar este modelo y ejecutar inferencias.

from sentence_transformers import SentenceTransformer

# Descargar desde el Hub de 🤗
model = SentenceTransformer("BookingCare/multilingual-embed-v1.0")
# Ejecutar inferencias
sentences = [
    'Chi phí điều trị xương khớp bằng tế bào gốc là bao nhiêu?',
    'Tôi muốn biết giá thành của phương pháp điều trị xương khớp bằng tế bào gốc.',
    'Bác sĩ nào giỏi về tim mạch ở Bệnh viện Tim Hà Nội?',
]
embeddings = model.encode(sentences)
print(embeddings.shape)
# [3, 768]

# Obtener las puntuaciones de similitud para las incrustaciones
similarities = model.similarity(embeddings, embeddings)
print(similarities.shape)
# [3, 3]