Views
No views yet
facebook/nllb-200-distilled-600M.1from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
2
3model_name = "Youme21/Manhwa-NLLB-EnAr-v1"
4
5tokenizer = AutoTokenizer.from_pretrained(model_name)
6model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
7
8text = " I must circulate my mana to heal my internal injuries."
9
10inputs = tokenizer(text, return_tensors="pt", padding=True)
11
12# Translate to Arabic (arb_Arab)
13translated_tokens = model.generate(
14 **inputs,
15 forced_bos_token_id=tokenizer.convert_tokens_to_ids("arb_Arab"),
16 max_length=128
17)
18
19result = tokenizer.batch_decode(translated_tokens, skip_special_tokens=True)[0]
20print(result)