Update README.md

README.md

@@ -60,9 +60,9 @@ def inference(text, model, tokenizer, max_input_tokens=1000, max_output_tokens=2
     return generated_text_answer
 
 
-print("Description input (test):", test_description)
+print("Description input (test):", text)
 
 print("Finetuned model's prompt: ")
-print(inference(test_description, model, tokenizer))
+print(inference(text, model, tokenizer))
 
 ```