@staticmethod
def get_pretrained_model(name, language, bucket=None):
    return BertSentenceEmbeddings.pretrained(name, language, bucket).setInputCols('sentence').setOutputCol('sentence_embeddings')
@staticmethod
def get_pretrained_model(name, language, bucket=None):
    return BertSentenceEmbeddings.pretrained(name, language, bucket).setInputCols('sentence').setOutputCol('sentence_embeddings')