sentence-transformers How to use sergioburdisso/dialog2flow-single-bert-base with sentence-transformers:
from sentence_transformers import SentenceTransformer
model = SentenceTransformer("sergioburdisso/dialog2flow-single-bert-base")
sentences = [
"your phone please",
"please get their phone number",
"okay can i get your phone number please to make that booking",
"okay can i please get your id number"
]
embeddings = model.encode(sentences)
similarities = model.similarity(embeddings, embeddings)
print(similarities.shape)
# [4, 4]