Update WordEmbeddingModel.scala