nairaxo commited on
Commit
5c7622e
1 Parent(s): ce5c462

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +5 -5
README.md CHANGED
@@ -29,7 +29,7 @@ You can use this model directly with a pipeline for masked language modeling:
29
 
30
  ```python
31
  >>> from transformers import pipeline
32
- >>> unmasker = pipeline('fill-mask', model='aioxlabs/bert-bantu')
33
  >>> unmasker("rais wa [MASK] ya tanzania.")
34
 
35
 
@@ -39,8 +39,8 @@ Here is how to use this model to get the features of a given text in PyTorch:
39
 
40
  ```python
41
  from transformers import BertTokenizer, BertModel
42
- tokenizer = BertTokenizer.from_pretrained('aioxlabs/bert-bantu')
43
- model = BertModel.from_pretrained("aioxlabs/bert-bantu")
44
  text = "Replace me by any text you'd like."
45
  encoded_input = tokenizer(text, return_tensors='pt')
46
  output = model(**encoded_input)
@@ -50,8 +50,8 @@ and in TensorFlow:
50
 
51
  ```python
52
  from transformers import BertTokenizer, TFBertModel
53
- tokenizer = BertTokenizer.from_pretrained('aioxlabs/bert-bantu')
54
- model = TFBertModel.from_pretrained("aioxlabs/bert-bantu")
55
  text = "Replace me by any text you'd like."
56
  encoded_input = tokenizer(text, return_tensors='tf')
57
  output = model(encoded_input)
 
29
 
30
  ```python
31
  >>> from transformers import pipeline
32
+ >>> unmasker = pipeline('fill-mask', model='nairaxo/bantu-bert')
33
  >>> unmasker("rais wa [MASK] ya tanzania.")
34
 
35
 
 
39
 
40
  ```python
41
  from transformers import BertTokenizer, BertModel
42
+ tokenizer = BertTokenizer.from_pretrained('nairaxo/bantu-bert')
43
+ model = BertModel.from_pretrained("nairaxo/bantu-bert")
44
  text = "Replace me by any text you'd like."
45
  encoded_input = tokenizer(text, return_tensors='pt')
46
  output = model(**encoded_input)
 
50
 
51
  ```python
52
  from transformers import BertTokenizer, TFBertModel
53
+ tokenizer = BertTokenizer.from_pretrained('nairaxo/bantu-bert')
54
+ model = TFBertModel.from_pretrained("nairaxo/bantu-bert")
55
  text = "Replace me by any text you'd like."
56
  encoded_input = tokenizer(text, return_tensors='tf')
57
  output = model(encoded_input)