abdulmatinomotoso commited on
Commit
15e4b11
·
1 Parent(s): 7f61988

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -10,8 +10,8 @@ labels = ['Politics', 'Tech', 'Entertainment', 'Business', 'World', 'Sport']
10
 
11
  #Defining the models and tokenuzer
12
  model_name = 'valurank/finetuned-distilbert-news-article-categorization'
13
- model = AutoModelForSequenceClassification.from_pretrained(model_name)
14
- tokenizer = AutoTokenizer.from_pretrained(model_name)
15
 
16
  #Reading in the text file
17
  def read_in_text(url):
 
10
 
11
  #Defining the models and tokenuzer
12
  model_name = 'valurank/finetuned-distilbert-news-article-categorization'
13
+ model = AutoModelForSequenceClassification.from_pretrained(model_name, use_auth_token=True)
14
+ tokenizer = AutoTokenizer.from_pretrained(model_name, use_auth_token=True)
15
 
16
  #Reading in the text file
17
  def read_in_text(url):