Hugging Face's logo

Spaces:
valurank
/
News_Articles_Categorization
Running

App Files Community
abdulmatinomotoso commited on
Commit
ac89df4
·
1 Parent(s): 7ff01e2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -7,8 +7,7 @@ from transformers import AutoTokenizer, AutoModelForSequenceClassification
7
  import torch
8
 
9
  #Defining the labels of the models
10
- labels = ['Politics', 'Tech', 'Entertainment', 'Business', 'World', 'Sport']
11
-
12
 
13
  #Defining the models and tokenuzer
14
  model_name = 'valurank/finetuned-distilbert-news-article-categorization'
 
7
  import torch
8
 
9
  #Defining the labels of the models
10
+ labels = ['health', 'entertainment', 'sport', 'politics', 'science', 'world', 'business', 'tech']
 
11
 
12
  #Defining the models and tokenuzer
13
  model_name = 'valurank/finetuned-distilbert-news-article-categorization'