Update README.md

README.md

@@ -8,6 +8,18 @@ datasets:
 
 ## Bert-base in Swahili
 
+This model was trained using HuggingFace's Flax framework and is part of the [JAX/Flax Community Week](https://discuss.huggingface.co/t/open-to-the-community-community-week-using-jax-flax-for-nlp-cv/7104) organized by [HuggingFace](https://huggingface.co). All training was done on a TPUv3-8 VM sponsored by the Google Cloud team.
+
+## How to use
+
+```python
+from transformers import AutoTokenizer, AutoModelForMaskedLM
+  
+tokenizer = AutoTokenizer.from_pretrained("flax-community/roberta-swahili")
+
+model = AutoModelForMaskedLM.from_pretrained("flax-community/roberta-swahili")
+```
+
 #### **Training Data**:
 This model was trained on [Swahili Safi](https://huggingface.co/datasets/flax-community/swahili-safi)