Views
No views yet
facebook/nllb-200-distilled-600M on East African AML transaction narratives.
Specialized for translating Luganda (lug_Latn) and Swahili (swh_Latn) mobile money
transaction descriptions to English for downstream AML classification.1from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
2
3model_id = "darthvader256/simitech-aml-afrinllb-translator"
4tokenizer = AutoTokenizer.from_pretrained(model_id)
5model = AutoModelForSeq2SeqLM.from_pretrained(model_id)
6
7tokenizer.src_lang = "lug_Latn"
8inputs = tokenizer("nkusaba ssente z'omusawo omukisa", return_tensors="pt")
9output = model.generate(
10 **inputs,
11 forced_bos_token_id=tokenizer.lang_code_to_id["eng_Latn"],
12 max_new_tokens=128,
13)
14print(tokenizer.decode(output[0], skip_special_tokens=True))
15# → "I am asking for doctor money, please"decision-plane/app/training/nlp_finetune.py — AfriNLLBTranslator class