Update README.md

README.md

@@ -42,7 +42,4 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
 model = AutoModelForCausalLM.from_pretrained("pAce576/llama3.2-1b-Instruct")
 tokenizer = AutoTokenizer.from_pretrained("pAce576/llama3.2-1b-Instruct")
 
-prompt = "Explain gravity to a 5-year-old."
-inputs = tokenizer(prompt, return_tensors="pt")
-outputs = model.generate(**inputs, max_new_tokens=100)
-print(tokenizer.decode(outputs[0], skip_special_tokens=True))
+#Your own generation function