Spaces:
Sleeping
Sleeping
padakantidilip commited on
Commit ·
72e29b7
1
Parent(s): 844d323
Add summarization app using BART model for some what longer texts
Browse files
app.py
CHANGED
|
@@ -5,7 +5,7 @@ from transformers import pipeline
|
|
| 5 |
summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
|
| 6 |
|
| 7 |
def summarize(text):
|
| 8 |
-
summary = summarizer(text, max_length=
|
| 9 |
return summary[0]['summary_text']
|
| 10 |
|
| 11 |
demo = gr.Interface(
|
|
|
|
| 5 |
summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
|
| 6 |
|
| 7 |
def summarize(text):
|
| 8 |
+
summary = summarizer(text, max_length=300, min_length=100, do_sample=False)
|
| 9 |
return summary[0]['summary_text']
|
| 10 |
|
| 11 |
demo = gr.Interface(
|