EE21 commited on
Commit
e377fb1
1 Parent(s): aa467bf

Update abstractive_summarization.py

Browse files
Files changed (1) hide show
  1. abstractive_summarization.py +1 -1
abstractive_summarization.py CHANGED
@@ -21,7 +21,7 @@ def summarize_with_bart_cnn(input_text):
21
  def summarize_with_led(input_text):
22
  pipe_led = pipeline("summarization", model="pszemraj/led-base-book-summary")
23
  summary = pipe_led(input_text, max_length=300, min_length=100, num_beams=1, early_stopping=False, length_penalty=1)
24
- return summary
25
 
26
  # Function to summarize using long-t5-tglobal-base-sci-simplify
27
  def summarize_with_t5(input_text):
 
21
  def summarize_with_led(input_text):
22
  pipe_led = pipeline("summarization", model="pszemraj/led-base-book-summary")
23
  summary = pipe_led(input_text, max_length=300, min_length=100, num_beams=1, early_stopping=False, length_penalty=1)
24
+ return summary[0]['summary_text']
25
 
26
  # Function to summarize using long-t5-tglobal-base-sci-simplify
27
  def summarize_with_t5(input_text):