Update 2023-04-21-leggen_flant5_base_en.md (#247)

JohnSnowLabs · May 22, 2023 · db78498 · db78498
1 parent ace8839
commit db78498
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/docs/_posts/gadde5300/2023-04-21-leggen_flant5_base_en.md b/docs/_posts/gadde5300/2023-04-21-leggen_flant5_base_en.md
@@ -44,7 +44,7 @@ document_assembler = nlp.DocumentAssembler()\
 
 flant5 = legal.TextGenerator.pretrained('leggen_flant5_base','en','legal/models')\
     .setInputCols(["question"])\
-    .setOutputCol("summary")
+    .setOutputCol("generated_text")
     .setMaxNewTokens(150)\
     .setStopAtEos(True)
 
@@ -53,7 +53,7 @@ data = spark.createDataFrame([
   [1, "Explain loan Clauses"]
 ]).toDF('id', 'text')
 results = pipeline.fit(data).transform(data)
-results.select("summary.result").show(truncate=False)
+results.select("generated_text.result").show(truncate=False)
 ```
 
 </div>