Commit
·
a45f613
1
Parent(s):
15e4b11
Update app.py
Browse files
app.py
CHANGED
|
@@ -8,6 +8,7 @@ import torch
|
|
| 8 |
#Defining the labels of the models
|
| 9 |
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, use_auth_token=True)
|
|
|
|
| 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)
|