Update README.md

README.md

@@ -30,7 +30,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
 model_name = "entfane/math-professor-3B"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForCausalLM.from_pretrained(model_name)
-input = "### User: Tell me some interesting fact \n### Assistant:"
+input = "### User: what is the derivative of 2x^2 \n### Assistant:"
 encoded_input = tokenizer(input, return_tensors = "pt").to(model.device)
 output = model.generate(**encoded_input, max_new_tokens=1024)
 print(tokenizer.decode(output[0], skip_special_tokens=False))