Update README.md

README.md

@@ -54,12 +54,12 @@ metrics:
 
   ```python
     def get_question(context, answer):
-        device = next(L_model.parameters()).device
+        device = next(QG_model.parameters()).device
         input_text = f"Given the context '{context}' and the answer '{answer}', what question can be asked?"
-        encoding = G_tokenizer.encode_plus(input_text, padding=True, return_tensors="pt").to(device)
+        encoding = QG_tokenizer.encode_plus(input_text, padding=True, return_tensors="pt").to(device)
     
-        output_tokens = L_model.generate(**encoding, early_stopping=True, num_beams=5, num_return_sequences=1, no_repeat_ngram_size=2, max_length=100)
-        out = G_tokenizer.decode(output_tokens[0], skip_special_tokens=True).replace("question:", "").strip()
+        output_tokens = QG_model.generate(**encoding, early_stopping=True, num_beams=5, num_return_sequences=1, no_repeat_ngram_size=2, max_length=100)
+        out = QG_tokenizer.decode(output_tokens[0], skip_special_tokens=True).replace("question:", "").strip()
     
         return out
   ```