Views
No views yet
1from transformers import MT5ForConditionalGeneration, MT5Tokenizer
2
3model_size = "base"
4model_name = f"persiannlp/mt5-{model_size}-parsinlu-opus-translation_fa_en"
5tokenizer = MT5Tokenizer.from_pretrained(model_name)
6model = MT5ForConditionalGeneration.from_pretrained(model_name)
7
8
9def run_model(input_string, **generator_args):
10 input_ids = tokenizer.encode(input_string, return_tensors="pt")
11 res = model.generate(input_ids, **generator_args)
12 output = tokenizer.batch_decode(res, skip_special_tokens=True)
13 print(output)
14 return output
15
16
17run_model("ستایش خدای را که پروردگار جهانیان است.")
18run_model("در هاید پارک کرنر بر گلدانی ایستاده موعظه میکند؛")
19run_model("وی از تمامی بلاگرها، سازمانها و افرادی که از وی پشتیبانی کردهاند، تشکر کرد.")
20run_model("مشابه سال ۲۰۰۱، تولید آمونیاک بی آب در ایالات متحده در سال ۲۰۰۰ تقریباً ۱۷،۴۰۰،۰۰۰ تن (معادل بدون آب) با مصرف ظاهری ۲۲،۰۰۰،۰۰۰ تن و حدود ۴۶۰۰۰۰۰ با واردات خالص مواجه شد. ")
21run_model("می خواهم دکترای علوم کامپیوتر راجع به شبکه های اجتماعی را دنبال کنم، چالش حل نشده در شبکه های اجتماعی چیست؟")['the admiration of God, which is the Lord of the world.']
['At the Ford Park, the Crawford Park stands on a vase;']
['He thanked all the bloggers, the organizations, and the people who supported him']
['similar to the year 2001, the economy of ammonia in the United States in the']
['I want to follow the computer experts on social networks, what is the unsolved problem in']['Adoration of God, the Lord of the world.']
['At the High End of the Park, Conrad stands on a vase preaching;']
['She thanked all the bloggers, organizations, and men who had supported her.']
['In 2000, the lack of water ammonia in the United States was almost']
['I want to follow the computer science doctorate on social networks. What is the unsolved challenge']['the praise of God, the Lord of the world.']
['At the Hyde Park Corner, Carpenter is preaching on a vase;']
['He thanked all the bloggers, organizations, and people who had supported him.']
['Similarly in 2001, the production of waterless ammonia in the United States was']
['I want to pursue my degree in Computer Science on social networks, what is the']