aditi2222 commited on
Commit
39ada04
·
1 Parent(s): 5cf8dd6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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(text) + ' </s>'
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,