Views
No views yet
1from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
2
3model_id = "morlayecis0003/nllb-200-pulaar-mairie"
4tokenizer = AutoTokenizer.from_pretrained(model_id)
5model = AutoModelForSeq2SeqLM.from_pretrained(model_id)
6
7text = "So a waylii kominaa, potaa anndinde kominaa ɓooyɗo on."
8inputs = tokenizer(text, return_tensors="pt")
9translated_tokens = model.generate(
10 **inputs,
11 forced_bos_token_id=tokenizer.convert_tokens_to_ids("fra_Latn"),
12 max_length=128
13)
14print(tokenizer.batch_decode(translated_tokens, skip_special_tokens=True)[0])