Update README.md

README.md

@@ -5,7 +5,7 @@ language:
 ---
 
 
-# oyo-bert
+# oyo-bert-base
 
 OYO-BERT (or Oyo-dialet of Yoruba BERT) was created by pre-training a [BERT model with token dropping](https://aclanthology.org/2022.acl-long.262/) on Yoruba language texts for about 100K steps. It was trained using BERT-base architecture
 
@@ -16,7 +16,7 @@ A mix of WURA, Wikipedia and MT560 Yoruba data
 You can use this model with Transformers *pipeline* for masked token prediction.
 ```python
 >>> from transformers import pipeline
->>> unmasker = pipeline('fill-mask', model='Davlan/oyo-bert')
+>>> unmasker = pipeline('fill-mask', model='Davlan/oyo-bert-base')
 >>> unmasker("Ọjọ kẹsan-an, [MASK] Kẹjọ ni wọn ri oku Baba")
 ```
 ```