Update README.md

README.md

@@ -17,20 +17,19 @@ from transformers import AutoTokenizer,AutoModelForSeq2SeqLM
 model = AutoModelForSeq2SeqLM.from_pretrained("Mihakram/Arabic_Question_Generation")
 tokenizer = AutoTokenizer.from_pretrained("Mihakram/Arabic_Question_Generation")
 
-def get_question(context,answer):
+def generate__questions(context,answer):
   text="context: " +context + " " + "answer: " + answer + " </s>"
   text_encoding = tokenizer.encode_plus(
       text,return_tensors="pt"
   )
   model.eval()
-  output = model.generate(
+  generated_ids =  model.generate(
     input_ids=text_encoding['input_ids'],
     attention_mask=text_encoding['attention_mask'],
     max_length=64,
     num_beams=5,
     num_return_sequences=1
   )
-
   return tokenizer.decode(generated_ids[0],skip_special_tokens=True,clean_up_tokenization_spaces=True).replace('question: ',' ')
 
 context="الثورة الجزائرية أو ثورة المليون شهيد، اندلعت في 1 نوفمبر 1954 ضد المستعمر الفرنسي ودامت 7 سنوات ونصف. استشهد فيها أكثر من مليون ونصف مليون جزائري"
@@ -61,4 +60,9 @@ If you want to cite this model you can use this:
 }
 ```
 
-> Created by [LinkedIn](https://www.linkedin.com/in/mihoubi-akram/)
+## Contacts
+
+**Mihoubi Akram Fawzi**: [Linkedin]() | [Twitter]() | [Github](https://github.com/mihoubi-akram) |<[email protected]>
+
+**Ibrir Adel**: [Linkedin]() | [Twitter]() | [Github]() |<>
+