Views
No views yet
1#Requirements !pip install transformers
2from transformers import AutoTokenizer,AutoModelForSeq2SeqLM
3
4model = AutoModelForSeq2SeqLM.from_pretrained("Mihakram/AraT5-base-question-generation")
5tokenizer = AutoTokenizer.from_pretrained("Mihakram/AraT5-base-question-generation")
6
7def get_question(context,answer):
8 text="context: " +context + " " + "answer: " + answer + " </s>"
9 text_encoding = tokenizer.encode_plus(
10 text,return_tensors="pt"
11 )
12 model.eval()
13 generated_ids = model.generate(
14 input_ids=text_encoding['input_ids'],
15 attention_mask=text_encoding['attention_mask'],
16 max_length=64,
17 num_beams=5,
18 num_return_sequences=1
19 )
20 return tokenizer.decode(generated_ids[0],skip_special_tokens=True,clean_up_tokenization_spaces=True).replace('question: ',' ')
21
22context="الثورة الجزائرية أو ثورة المليون شهيد، اندلعت في 1 نوفمبر 1954 ضد المستعمر الفرنسي ودامت 7 سنوات ونصف. استشهد فيها أكثر من مليون ونصف مليون جزائري"
23answer =" 7 سنوات ونصف"
24
25get_question(context,answer)
26
27#output : question="كم استمرت الثورة الجزائرية؟ "
28