Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -25,7 +25,7 @@ model = BartForConditionalGeneration.from_pretrained("facebook/bart-large-cnn")
|
|
25 |
tokenizer = BartTokenizer.from_pretrained("facebook/bart-large-cnn")
|
26 |
|
27 |
def article_summarization(result):
|
28 |
-
input_ = str(
|
29 |
# generate summary
|
30 |
input_ids = tokenizer.encode(result, return_tensors='pt')
|
31 |
summary_ids = model.generate(input_ids,
|
|
|
25 |
tokenizer = BartTokenizer.from_pretrained("facebook/bart-large-cnn")
|
26 |
|
27 |
def article_summarization(result):
|
28 |
+
input_ = str(result) + ' </s>'
|
29 |
# generate summary
|
30 |
input_ids = tokenizer.encode(result, return_tensors='pt')
|
31 |
summary_ids = model.generate(input_ids,
|