Update README.md

README.md

@@ -2,6 +2,7 @@
 library_name: transformers
 tags: []
 ---
+```
 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer, TorchAoConfig
 
@@ -37,4 +38,5 @@ generated_ids = quantized_model.generate(**inputs, max_new_tokens=128)
 output_text = tokenizer.batch_decode(
     generated_ids, skip_special_tokens=True, clean_up_tokenization_spaces=False
 )
-print("Response:", output_text[0][len(prompt) :])
+print("Response:", output_text[0][len(prompt) :])
+```