EuropeanParliament
/

eurovoc_eu

Text Classification

Inference Endpoints

Model card Files Files and versions Community

scampion commited on Nov 17, 2023

Commit

2e88644

•

1 Parent(s): 33116db

Update handler.py

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -8,7 +8,7 @@ import torch
 from eurovoc import EurovocTagger
-BERT_MODEL_NAME = "nlpaueb/legal-bert-base-uncased"
 MAX_LEN = 512
 TEXT_MAX_LEN = MAX_LEN * 50
 tokenizer = AutoTokenizer.from_pretrained(BERT_MODEL_NAME)

 from eurovoc import EurovocTagger
+BERT_MODEL_NAME = "EuropeanParliament/EUBERT"
 MAX_LEN = 512
 TEXT_MAX_LEN = MAX_LEN * 50
 tokenizer = AutoTokenizer.from_pretrained(BERT_MODEL_NAME)