Spaces:
Sleeping
Sleeping
Update abstractive_summarization.py
Browse files
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):
|