--- license: apache-2.0 language: - yo - en metrics: - rouge pipeline_tag: translation tags: - text - machine-translation - language-translation - seq2seq - helsinki-nlp --- ## Predicting English Translation ```python from transformers import AutoTokenizer, AutoModelForSeq2SeqLM # Loading tokenizer and model tokenizer = AutoTokenizer.from_pretrained("kingabzpro/Helsinki-NLP-opus-yor-mul-en") model = AutoModelForSeq2SeqLM.from_pretrained("kingabzpro/Helsinki-NLP-opus-yor-mul-en").to('cuda') # Prediction a = model.generate(**tokenizer.prepare_seq2seq_batch('Nínú ìpè kan lẹ́yìn ìgbà náà, wọ́n sọ fún aṣojú iléeṣẹ́ BlaBlaCar pé ètò náà ti yí padà, pé',return_tensors='pt').to('cuda')) text = tokenizer.batch_decode(a) # Cleaning text text = str(text) text = re.sub(" ","",text) text = re.sub("'","",text) text = text.replace("[", "") text = text.replace("]", "") text ``` ## Result ``` 'In a statement after that hearing, the BualaCard’s representative was told that the event had changed, that he had turned up.' ``` ## ROGUE Score **0.3025**