nikhedward commited on
Commit
d17e92a
1 Parent(s): 5bd2a5a

added correct model

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -40,9 +40,9 @@ Abstractive Summarization: The abstractive approach involves rephrasing the comp
40
 
41
 
42
 
43
- model_name = 'facebook/bart-large-cnn'
44
- tokenizer = BartTokenizer.from_pretrained(model_name)
45
- model = BartForConditionalGeneration.from_pretrained(model_name)
46
 
47
  def summarize(inp):
48
  inp = inp.replace('\n','')
 
40
 
41
 
42
 
43
+ model_name = "huggingface/nikhedward/bart-large-cnn-finetuned-multi-news"
44
+ tokenizer = AutoTokenizer.from_pretrained(model_name)
45
+ model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
46
 
47
  def summarize(inp):
48
  inp = inp.replace('\n','')