Update README.md
Browse files
README.md
CHANGED
@@ -172,7 +172,7 @@ model_id = "TurkuNLP/multilingual-web-register-classification"
|
|
172 |
|
173 |
# Load model and tokenizer
|
174 |
model = AutoModelForSequenceClassification.from_pretrained(model_id).to(device)
|
175 |
-
tokenizer = AutoTokenizer.from_pretrained(
|
176 |
|
177 |
# Text to be categorized
|
178 |
text = "A text to be categorized"
|
|
|
172 |
|
173 |
# Load model and tokenizer
|
174 |
model = AutoModelForSequenceClassification.from_pretrained(model_id).to(device)
|
175 |
+
tokenizer = AutoTokenizer.from_pretrained("xlm-roberta-large")
|
176 |
|
177 |
# Text to be categorized
|
178 |
text = "A text to be categorized"
|