Update README.md

README.md

@@ -1,6 +1,6 @@
 # how to use 
 
-
+```
 messages = [
     {"role": "system", "content": system_prompt},
     {"role": "user",   "content": "What is the sqrt of 101?"},
@@ -19,6 +19,7 @@ _ = model.generate(
     temperature = 1.0, top_p = 0.95, top_k = 64,
     streamer = TextStreamer(tokenizer, skip_prompt = True),
 )
+```
 
 # expected output