beshiribrahim commited on
Commit
df41dd2
·
verified ·
1 Parent(s): 4681759

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -3
README.md CHANGED
@@ -33,11 +33,10 @@ The model is designed for **masked language modeling** — predicting missing wo
33
  ### Example (Python)
34
 
35
  ```python
36
- from transformers import AutoTokenizer, AutoModelForMaskedLM
37
- import torch
38
 
39
  tokenizer = AutoTokenizer.from_pretrained("BeitTigreAI/tigre-xlm-roberta-base")
40
- model = AutoModelForMaskedLM.from_pretrained("BeitTigreAI/tigre-xlm-roberta-base")
41
 
42
  text = "[tig_Ethi] መርሐበ ብኩም እት <mask>"
43
 
 
33
  ### Example (Python)
34
 
35
  ```python
36
+ from transformers import AutoTokenizer, XLMRobertaForMaskedLM
 
37
 
38
  tokenizer = AutoTokenizer.from_pretrained("BeitTigreAI/tigre-xlm-roberta-base")
39
+ model = XLMRobertaForMaskedLM.from_pretrained("BeitTigreAI/tigre-xlm-roberta-base")
40
 
41
  text = "[tig_Ethi] መርሐበ ብኩም እት <mask>"
42