all-MiniLM-L6-v2 converted for SparkNLP

def embedSentences(sentences: DataFrame): DataFrame = {

    val documentAssembler = new DocumentAssembler()
      .setInputCol("sentence")
      .setOutputCol("document")

    val tokenizer = new Tokenizer()
      .setInputCols("document")
      .setOutputCol("token")

        val embeddings = BertEmbeddings
          .pretrained("all_minilm_l6", "en")
          .setInputCols(Array("document", "token"))
          .setOutputCol("embeddings")


    val sentenceEmbeddings = new SentenceEmbeddings()
      .setInputCols("document", "embeddings")
      .setOutputCol("sentence_embeddings")
      .setPoolingStrategy("AVERAGE")

    val pipeline = new Pipeline().setStages(Array(
      documentAssembler,
      tokenizer,
      embeddings,
      sentenceEmbeddings
    ))

    val pipelineModel = pipeline.fit(sentences)
    val pipelineDF = pipelineModel.transform(sentences)