Update README.md

README.md

@@ -85,7 +85,8 @@ Users (both direct and downstream) should be made aware of the risks, biases and
 
 ## How to Get Started with the Model
 In Google Colab:
-'''
+```
+
 !pip install -q -U transformers peft accelerate optimum
 !pip install datasets==2.15.0
 !pip install auto-gptq --extra-index-url https://huggingface.github.io/autogptq-index/whl/cu117/
@@ -124,7 +125,8 @@ prompt_template=f'''<s>[INST] {prompt} [/INST]'''
 input_ids = tokenizer(prompt, return_tensors='pt').input_ids.cuda()
 output = persisted_model.generate(inputs=input_ids, temperature=0.7, do_sample=True, top_p=0.95, top_k=40, max_new_tokens=512)
 print(tokenizer.decode(output[0]))    
-'''
+
+```
 
 # To perform inference on the test dataset example load the model from the checkpoint
 persisted_model = AutoPeftModelForCausalLM.from_pretrained(