Views
No views yet
1from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
2
3model_name = "kelSidenna/nllb-eng-ha-v0"
4
5tokenizer = AutoTokenizer.from_pretrained(model_name)
6model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
7
8text = "How are you?"
9tokenizer.src_lang = "en"
10inputs = tokenizer(text, return_tensors="pt")
11translated_tokens = model.generate(
12 **inputs,
13 forced_bos_token_id=tokenizer.convert_tokens_to_ids("ha-ar")
14)
15print(tokenizer.batch_decode(translated_tokens, skip_special_tokens=True))
16