erikhenriksson commited on
Commit
a22ad8b
·
verified ·
1 Parent(s): a111457

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
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(model_id)
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"