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]
