English-Yoruba STEM Translation Model
This model is trained to translate English STEM content to Yoruba.
Model Details
- Architecture: Transformer-based sequence-to-sequence model
- Base Model: Davlan/mt5-base-en-yor-mt
- Training Data: YorubaSTEM1.0
- Performance: BLEU: 36.08
Usage
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
tokenizer = AutoTokenizer.from_pretrained("gbelewade/YorubaSTEMt5")
model = AutoModelForSeq2SeqLM.from_pretrained("gbelewade/YorubaSTEMt5")
Translate English text to Yoruba
english_text = "The chemical formula for water is H2O."
inputs = tokenizer(english_text, return_tensors="pt")
outputs = model.generate(**inputs)
yoruba_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
print(yoruba_text)
Limitations
[Describe any known limitations of the model]
Citation
