Update README.md

README.md

@@ -49,7 +49,7 @@ model = AutoModelForCausalLM.from_pretrained("Praneeth/code-gemma-2b-it", device
 input_text = "Write me a poem about Machine Learning."
 input_ids = tokenizer(input_text, return_tensors="pt").to("cuda")
 
-outputs = model.generate(**input_ids)
+outputs = model.generate(**input_ids, max_new_tokens=256,)
 print(tokenizer.decode(outputs[0]))
 ```