Update README.md

README.md

@@ -41,12 +41,12 @@ class QuestionGeneration:
         )
         input_ids = encoding['input_ids'].to(self.device)
         attention_mask = encoding['attention_mask'].to(self.device)
-        beam_outputs = self.model.generate(
+        outputs = self.model.generate(
             input_ids = input_ids,
             attention_mask = attention_mask
         )
         question = self.tokenizer.decode(
-            beam_outputs[0],
+            outputs[0],
             skip_special_tokens = True,
             clean_up_tokenization_spaces = True
         )