Update README.md

README.md

@@ -39,7 +39,7 @@ text = "The model was fine-tuned using knowledge distillation techniques. The tr
 prompt = f"Translate input sentence to Korean \n### Input: {text} \n### Translated:"
 
 # Tokenize and generate translation
-input_ids = tokenizer(message_0, return_tensors="pt").to(model.device)
+input_ids = tokenizer(prompt, return_tensors="pt").to(model.device)
 outputs = model.generate(**input_ids, max_new_tokens=1024)
 out_message = tokenizer.decode(outputs[0][len(input_ids["input_ids"][0]):], skip_special_tokens=True)