sentence-transformers How to use vany02/hitachi-defect-classifier with sentence-transformers:
from sentence_transformers import SentenceTransformer
model = SentenceTransformer("vany02/hitachi-defect-classifier")
sentences = [
"1位側開戸排水ﾊﾟﾝ内 切粉(最大3mm×5ヶ)",
"切粉・素線など伝導性の高い異物。この場合はサイズおよび個数を記録する。",
"ひっかき傷に類するｷｽﾞがあるもの",
"塗装の色むら・色違い等。ただし、塗装自体が未了の場合は「503作業漏れ」に分類する。"
]
embeddings = model.encode(sentences)
similarities = model.similarity(embeddings, embeddings)
print(similarities.shape)
# [4, 4] 
