Update README.md

README.md

@@ -19,10 +19,13 @@ Import the scripts from the code folder
 from model import Mamba, ModelArgs
 ```
 
-Loading Model```
+
+Loading Model
+```
 mamba_model = Mamba.from_pretrained("pt-sk/mamba").to("cuda")
 ```
 
+
 Loading Tokenizer
 ```
 tokenizer = AutoTokenizer.from_pretrained('pt-sk/mamba')