Update README.md

README.md

@@ -68,13 +68,26 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
 tokenizer = AutoTokenizer.from_pretrained("kaiiddo/A3ON-1B")
 model = AutoModelForCausalLM.from_pretrained("kaiiddo/A3ON-1B")
 
-# Generate text
+# Set pad_token_id to eos_token_id to avoid warnings
+model.config.pad_token_id = model.config.eos_token_id
+
+# Generate text with adjusted parameters
 inputs = tokenizer("Hello, how can I help you today?", return_tensors="pt")
-outputs = model.generate(**inputs, max_length=500)
+outputs = model.generate(
+    **inputs,
+    max_length=500,
+    do_sample=True,
+    temperature=0.7,
+    top_k=50
+)
+
+# Decode the output and split into lines
 response = tokenizer.decode(outputs[0], skip_special_tokens=True)
+response_lines = response.split('\n')
 
-# Print the response
-print(response)
+# Print each line of the response
+for line in response_lines:
+    print(line)
 ```
 
 ### Model Parameter Count