Update README.md

README.md

@@ -2,8 +2,9 @@
 license: apache-2.0
 ---
 
+Here is a code to create this tiny model:
 
-# === Obtained with following code. ===
+```python
 import os
 import torch
 torch.set_default_dtype(torch.bfloat16)
@@ -28,4 +29,4 @@ output_dir = "./tiny-random-aya-base/"
 os.makedirs(output_dir, exist_ok=True)
 model.save_pretrained(output_dir, safe_serialization=True)
 tokenizer.save_pretrained(output_dir)
-
+```