diff --git a/docs/_posts/Cabir40/2023-03-01-t5_flan_base_xx.md b/docs/_posts/Cabir40/2023-03-01-t5_flan_base_xx.md index a62fda8156585a..f0641bc33902a3 100644 --- a/docs/_posts/Cabir40/2023-03-01-t5_flan_base_xx.md +++ b/docs/_posts/Cabir40/2023-03-01-t5_flan_base_xx.md @@ -50,8 +50,8 @@ result = pipeline.fit(data).transform(data) ``` ```scala val documentAssembler = new DocumentAssembler() -.setInputCols("text") -.setOutputCols("document") +.setInputCol("text") +.setOutputCol("document") val t5 = T5Transformer.pretrained("t5_flan_base","xx") .setInputCols("document") @@ -81,4 +81,4 @@ val result = pipeline.fit(data).transform(data) ## References -https://huggingface.co/google/flan-t5-base \ No newline at end of file +https://huggingface.co/google/flan-t5-base