TurkuNLP
/

web-register-classification-multilingual

Text Classification

Model card Files Files and versions

erikhenriksson commited on Mar 18

Commit

a22ad8b

·

verified ·

1 Parent(s): a111457

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -172,7 +172,7 @@ model_id = "TurkuNLP/multilingual-web-register-classification"
 # Load model and tokenizer
 model = AutoModelForSequenceClassification.from_pretrained(model_id).to(device)
-tokenizer = AutoTokenizer.from_pretrained(model_id)
 # Text to be categorized
 text = "A text to be categorized"

 # Load model and tokenizer
 model = AutoModelForSequenceClassification.from_pretrained(model_id).to(device)
+tokenizer = AutoTokenizer.from_pretrained("xlm-roberta-large")
 # Text to be categorized
 text = "A text to be categorized"