Update README.md

README.md

@@ -50,6 +50,7 @@ print(res)
 
 Alternatively, if you prefer to not use trust_remote_code=True you can download instruct_pipeline.py, store it alongside your notebook, and construct the pipeline yourself from the loaded model and tokenizer:
 
+```python
 from instruct_pipeline import InstructionTextGenerationPipeline
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
@@ -58,3 +59,4 @@ tokenizer = AutoTokenizer.from_pretrained("aisquared/dlite-v2-1_5b", padding_sid
 model = AutoModelForCausalLM.from_pretrained("aisquared/dlite-v2-1_5b", device_map="auto", torch_dtype=torch.bfloat16)
 
 generate_text = InstructionTextGenerationPipeline(model=model, tokenizer=tokenizer)
+```