EE21 commited on
Commit
c09e1a9
1 Parent(s): fb5ed70

Update abstractive_summarization.py

Browse files
Files changed (1) hide show
  1. abstractive_summarization.py +1 -1
abstractive_summarization.py CHANGED
@@ -19,7 +19,7 @@ def summarize_with_bart_cnn(input_text):
19
 
20
  # Function to summarize using led-base-book-summary
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
 
 
19
 
20
  # Function to summarize using led-base-book-summary
21
  def summarize_with_led(input_text):
22
+ pipe_led = pipeline("summarization", model="pszemraj/led-large-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