abdulmatinomotoso commited on
Commit
4c4e6d4
1 Parent(s): 174033e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -36,9 +36,9 @@ from transformers import (
36
  )
37
 
38
  #initializing the tokenizer and the model
39
- model_type_2 ="chinhon/pegasus-multi_news-headline"
40
- tokenizer_2 = AutoTokenizer.from_pretrained(model_type_2)
41
- model_2 = AutoModelForSeq2SeqLM.from_pretrained(model_type_2)
42
 
43
  #Defining a function to generate the headlines
44
  def headline_generator_2(file):
 
36
  )
37
 
38
  #initializing the tokenizer and the model
39
+ model_type_2 ="valurank/pegasus-multi_news-headline"
40
+ tokenizer_2 = AutoTokenizer.from_pretrained(model_type_2, use_auth_token='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')
41
+ model_2 = AutoModelForSeq2SeqLM.from_pretrained(model_type_2, use_auth_token='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')
42
 
43
  #Defining a function to generate the headlines
44
  def headline_generator_2(file):