Update 2023-04-21-fingen_flant5_base_en.md (#246)

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