kabita-choudhary commited on
Commit
74c4cd3
·
1 Parent(s): 20e4075

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -14,7 +14,7 @@ def summarize(data, modelname):
14
  model = PegasusForConditionalGeneration.from_pretrained("google/pegasus-xsum")
15
  tokenizer = PegasusTokenizer.from_pretrained("google/pegasus-xsum")
16
  else:
17
- summarizer = pipeline("summarization", model="philschmid/bart-large-cnn-samsum")
18
  print("world")
19
  output = summarizer(data, max_length=300, min_length=30, do_sample=False)
20
  return output[0]["summary_text"]
 
14
  model = PegasusForConditionalGeneration.from_pretrained("google/pegasus-xsum")
15
  tokenizer = PegasusTokenizer.from_pretrained("google/pegasus-xsum")
16
  else:
17
+ summarizer = pipeline("summarization", model="lidiya/bart-large-xsum-samsum")
18
  print("world")
19
  output = summarizer(data, max_length=300, min_length=30, do_sample=False)
20
  return output[0]["summary_text"]