Update README.md

README.md

@@ -50,18 +50,14 @@ Download all files from the repository into a single folder and run the notebook
 ### Loading the Model
 
 ```python
+import json
 import torch
 from safetensors import safe_open
 from model import Transformer as Model
 from transformers import PreTrainedTokenizerFast
 
-config = {
-    'layers': 24,
-    'num_heads': 12,
-    'vocab_size': 32768,
-    'input_dims': 768,
-    'hidden_dims': 3072,
-}
+with open("./config.json", "r") as f:
+    config = json.load(f)
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
 torch.set_default_device(device)