abdulmatinomotoso commited on
Commit
5273dc1
·
1 Parent(s): ac89df4

Update app.py

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