Update README.md

README.md

@@ -37,9 +37,9 @@ text = "answer:Saint Bernadette Soubirous content:Architecturally , the school h
 raw_input_ids = tokenizer.encode(text)
 input_ids = [tokenizer.bos_token_id] + raw_input_ids + [tokenizer.eos_token_id]
 
-summary_ids = model.generate(torch.tensor([input_ids]))
+question_ids = model.generate(torch.tensor([input_ids]))
 
-decode = tokenizer.decode(summary_ids.squeeze().tolist(), skip_special_tokens=True)
+decode = tokenizer.decode(question_ids.squeeze().tolist(), skip_special_tokens=True)
 
 decode = decode.replace(' # # ', '').replace('  ', ' ').replace(' ##', '')