huudan123/model_stage4

Uso Directo (Transformadores de Oraciones)

from sentence_transformers import SentenceTransformer

# Descargar desde el Hub de 🤗
model = SentenceTransformer("huudan123/model_stage4")
# Ejecutar inferencia
sentences = [
    'Một người đàn ông đang lắp ráp các bộ phận loa.',
    'Một người đàn ông đang đi bộ trên vỉa hè.',
    'Một người đàn ông phun nước từ vòi cho một người đàn ông khác.',
]
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]