Update abstractive_summarization.py

abstractive_summarization.py

@@ -15,7 +15,7 @@ def summarize_with_bart_ft(input_text):
 def summarize_with_bart_cnn(input_text):
     pipe = pipeline("summarization", model="facebook/bart-large-cnn")
     summary = pipe(input_text, max_length=300, min_length=100, num_beams=1, early_stopping=False, length_penalty=1)
-    return summary[0]['summary_text']
+    return summary
 
 # Function to summarize using led-base-book-summary
 def summarize_with_led(input_text):