Downloads · 30 days
12
32% of all-time downloads
Dims002/mbart-yemba-translator
mbart-yemba-translator is a translation model from Dims002. Use it when you need text moved from one language to another. The card lists the license as apache-2.0.
Ce modèle est un fine-tuning de mBART pour la traduction entre le français et le yemba.
Downloads · 30 days
12
32% of all-time downloads
All-time downloads
38
Public
Parameters
611M
7.4 GB on disk
Likes
0
Public
Click a slice to open those files.
.pt4.9 GB · 66%
From the Hugging Face model README
Ce modèle est un fine-tuning de mBART pour la traduction entre le français et le yemba.
Voici comment utiliser le modèle:
from transformers import MBartForConditionalGeneration, MBart50TokenizerFast
model = MBartForConditionalGeneration.from_pretrained("Dims002/mbart-yemba-translator") tokenizer = MBart50TokenizerFast.from_pretrained("Dims002/mbart-yemba-translator")
def translate_text(text, src_lang="fr_XX", tgt_lang="yemba"): tokenizer.src_lang = src_lang encoded = tokenizer(text, return_tensors="pt") generated_tokens = model.generate(**encoded) return tokenizer.batch_decode(generated_tokens, skip_special_tokens=True)[0]