Views
No views yet
1from transformers import MT5ForConditionalGeneration, MT5Tokenizer
2
3model_size = "small"
4model_name = f"persiannlp/mt5-{model_size}-parsinlu-multiple-choice"
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("وسیع ترین کشور جهان کدام است؟ <sep> آمریکا <sep> کانادا <sep> روسیه <sep> چین")
18run_model("طامع یعنی ؟ <sep> آزمند <sep> خوش شانس <sep> محتاج <sep> مطمئن")
19run_model(
20 "زمینی به ۳۱ قطعه متساوی مفروض شده است و هر روز مساحت آماده شده برای احداث، دو برابر مساحت روز قبل است.اگر پس از (۵ روز) تمام زمین آماده شده باشد، در چه روزی یک قطعه زمین آماده شده <sep> روز اول <sep> روز دوم <sep> روز سوم <sep> هیچکدام")
21