BGE base Financial Matryoshka

Primero, instala la librería Sentence Transformers:

pip install -U sentence-transformers

Luego puedes cargar este modelo y ejecutar inferencias.

from sentence_transformers import SentenceTransformer

# Descargar del Hub de 🤗
model = SentenceTransformer("adriansanz/sitges2608")
# Ejecutar inferencia
sentences = [
'El termini per a la presentació de les sol·licituds de modificació del projecte o activitat subvencionat és de 15 dies naturals abans de la finalització del projecte o activitat.',
'Quin és el termini per a la presentació de les sol·licituds de modificació del projecte o activitat subvencionat?',
"Quin és el registre on es troben les dades d'inscripció?",
]
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]