Hugging Face's logo

Spaces:
valurank
/
News_Articles_Categorization
Running

App Files Community
abdulmatinomotoso commited on
Commit
1073c6d
·
1 Parent(s): a45f613

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ import torch
8
  #Defining the labels of the models
9
  labels = ['Politics', 'Tech', 'Entertainment', 'Business', 'World', 'Sport']
10
 
11
- !transformers-cli login
12
  #Defining the models and tokenuzer
13
  model_name = 'valurank/finetuned-distilbert-news-article-categorization'
14
  model = AutoModelForSequenceClassification.from_pretrained(model_name, use_auth_token=True)
 
8
  #Defining the labels of the models
9
  labels = ['Politics', 'Tech', 'Entertainment', 'Business', 'World', 'Sport']
10
 
11
+
12
  #Defining the models and tokenuzer
13
  model_name = 'valurank/finetuned-distilbert-news-article-categorization'
14
  model = AutoModelForSequenceClassification.from_pretrained(model_name, use_auth_token=True)