Update README.md

README.md

@@ -17,17 +17,18 @@ The model's output may be influenced by biases present in the training data, pot
 
 ```python
 
+#!pip install transformers
+from transformers import AutoTokenizer, GPT2LMHeadModel
 checkpoint = "AbdelrahmanFakhry/finetuned-gpt2-multi-QA-Generation"
 
 tokenizer = AutoTokenizer.from_pretrained(checkpoint)
 model = GPT2LMHeadModel.from_pretrained(checkpoint)
 # Retrieve a test question from the test dataset
-test_text = test_dataset.to_dict()['question'][3]
-test_text
+#test_text = test_dataset.to_dict()['question'][3]
 
 
 # test_text should be like that 
-'''Below is input text, the task is to generate questions from input text and multiple answers for 
+test_text = '''Below is input text, the task is to generate questions from input text and multiple answers for 
 each question and provide hint and correct answer for each question.\n\n### Input:\n<hl> Local intercellular
 communication is the province of the paracrine , also called a paracrine factor , which is a chemical that
 induces a response in neighboring cells . <hl> Although paracrines may enter the bloodstream , their concentration 
@@ -76,8 +77,7 @@ def inference(text, model, tokenizer, max_input_tokens=3000, max_output_tokens=5
     return generated_text_answer
 
 generated_answer = inference(test_text, model, tokenizer)
-print('Generated Answer:')
-print(generated_answer)
+#Generated Answer should be look like that:
 '''
 "Choices': ['paracrine factor', 'paracrine factor', 'paracrine factor II', 'paracrine factor III'], 
 'Question': 'Which of the following is not a paracrine factor?',
@@ -85,6 +85,8 @@ print(generated_answer)
 'hint': 'Local intercellular communication is the province of the paracrine, also called a paracrine factor,
 which is a chemical that induces a response in neighboring cells."
 '''
+print('Generated Answer:')
+print(generated_answer)
 ```
 
 # Acknowledgments