abdulmatinomotoso
commited on
Commit
·
15e4b11
1
Parent(s):
7f61988
Update app.py
Browse files
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):
|