Views
No views yet

1from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
2
3tokenizer = AutoTokenizer.from_pretrained("Ijwi-ry-Ikirundi-AI/nllb-kirundi-multi")
4model = AutoModelForSeq2SeqLM.from_pretrained("Ijwi-ry-Ikirundi-AI/nllb-kirundi-multi")
5
6# Example: Kirundi -> French
7tokenizer.src_lang = "run_Latn"
8text = "Amahoro y'Imana abane nawe."
9
10inputs = tokenizer(text, return_tensors="pt")
11output = model.generate(
12 **inputs,
13 forced_bos_token_id=tokenizer.convert_tokens_to_ids("fra_Latn")
14)
15
16print(tokenizer.decode(output[0], skip_special_tokens=True))