bonadossou commited on
Commit
133eaf3
1 Parent(s): 3caab80

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -3
README.md CHANGED
@@ -78,9 +78,9 @@ Model | MasakhaNER | MasakhaNER2.0* | Text Classification (Yoruba/Hausa) | Senti
78
 
79
  ## HuggingFace usage of AfroLM-large
80
  ```python
81
- from transformers import AutoTokenizer, AutoModelForTokenClassification
82
- model = AutoModelForTokenClassification.from_pretrained("bonadossou/afrolm_active_learning")
83
- tokenizer = AutoTokenizer.from_pretrained("bonadossou/afrolm_active_learning")
84
  tokenizer.model_max_length = 256
85
  ```
86
 
 
78
 
79
  ## HuggingFace usage of AfroLM-large
80
  ```python
81
+ from transformers import XLMRobertaModel, XLMRobertaTokenizer
82
+ model = XLMRobertaModel.from_pretrained("bonadossou/afrolm_active_learning")
83
+ tokenizer = XLMRobertaTokenizer.from_pretrained("bonadossou/afrolm_active_learning")
84
  tokenizer.model_max_length = 256
85
  ```
86