nllb-200-darjia-en

Moroccan Darija → English Translation (NLLB-200)

This model is a fine-tuned version of facebook/nllb-200-distilled-600m for machine translation from Moroccan Darija to English.

It is intended for informal, conversational Darija rather than Modern Standard Arabic.

Languages

• Source: Moroccan Darija (ary_Arab)
• Target: English (eng_Latn)
• Primary script: Latin

Training Data

The model was fine-tuned on a custom English–Darija parallel dataset compiled from the Darija Open Dataset.

Usage

from transformers import AutoTokenizer, AutoModelForSeq2SeqLM

model_name = "mwkhettab/nllb-200-darija-en"

tokenizer = AutoTokenizer.from_pretrained(model_name)
model = AutoModelForSeq2SeqLM.from_pretrained(model_name)

text = "شنو الطقس دابا برا؟"

inputs = tokenizer(text, return_tensors="pt", max_length=512)

outputs = model.generate(
    **inputs,
    forced_bos_token_id=tokenizer.convert_tokens_to_ids("eng_Latn"),
    max_new_tokens=128,
    num_beams=5,
)

translation = tokenizer.decode(
    outputs[0],
    skip_special_tokens=True,
).strip()

print(translation)