Views
No views yet
facebook/bart-base pour la traduction du Wolof vers le Français.facebook/bart-base1from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
2
3tokenizer = AutoTokenizer.from_pretrained("Diamweli/wolof-bart-finetuned")
4model = AutoModelForSeq2SeqLM.from_pretrained("Diamweli/wolof-bart-finetuned")
5
6# Exemple de traduction
7text = "Ambedkar mingi judd ci xeetu Dalit, di xeet wu ñu joxul benn cër ci Inde ndax dañu naan kenn du leen laal."
8inputs = tokenizer(text, return_tensors="pt")
9outputs = model.generate(
10 **inputs,
11 max_new_tokens=100,
12 num_beams=4,
13 early_stopping=True,
14)
15result = tokenizer.decode(outputs[0], skip_special_tokens=True)
16print(result)
17# → Traduction en françaissrc_max_len = 74, tgt_max_len = 99| Metric | Score |
|---|---|
| BLEU | 4.8945 |
| ROUGE-1 | 0.2454 |
| ROUGE-2 | 0.0745 |
| ROUGE-L | 0.2098 |
| ROUGE-Lsum | 0.2097 |
| Test Loss | 3.5154 |