Update README.md

README.md

@@ -120,6 +120,29 @@ MiniAxion1 is not intended to compete with large-scale models. Instead, it is bu
 
 ---
 
+### Quick start
+
+```python
+
+import torch
+from model import NRMModel
+from tokenizer import Tokenizer
+
+# load
+model = NRMModel.from_config("config.json")
+model.load_state_dict(torch.load("model.pt"))
+model.eval()
+
+tokenizer = Tokenizer.load("tokenizer.json")
+
+def generate(prompt):
+    tokens = tokenizer.encode(prompt)
+    output = model.generate(tokens)
+    return tokenizer.decode(output)
+
+print(generate("<INST>What is 2 + 2?</INST>"))
+```
+
 ## 🧠 Philosophy
 
 MiniAxion1 explores a key question: