Update README.md

README.md

@@ -11,7 +11,7 @@ This is a runnable standalone huggingface implementation for one of the models.
 Some trivial code to load the locally converted HF model + tokenizer and
 run a tiny generation.
 
-
+```
 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
 
@@ -42,7 +42,7 @@ if __name__ == "__main__":
     print(PROMPT)
     print("\n=== Generation ===")
     print(tok.decode(out[0], skip_special_tokens=True))
-
+```
 
 
 ## License