Update README.md

README.md

@@ -31,7 +31,7 @@ def get_question(context,answer):
     num_return_sequences=1
   )
 
-  return tokenizer.decode(output[0],skip_special_tokens=True,clean_up_tokenization_spaces=True)
+  return tokenizer.decode(generated_ids[0],skip_special_tokens=True,clean_up_tokenization_spaces=True).replace('question: ',' ')
 
 context="الثورة الجزائرية أو ثورة المليون شهيد، اندلعت في 1 نوفمبر 1954 ضد المستعمر الفرنسي ودامت 7 سنوات ونصف. استشهد فيها أكثر من مليون ونصف مليون جزائري"
 answer =" 7 سنوات ونصف"