Update README.md

README.md

@@ -56,7 +56,7 @@ Users (both direct and downstream) should be made aware of the risks, biases and
 
 ## How to Get Started with the Model
 
-'''
+```
 
 import torch
 from peft import PeftModel, PeftConfig
@@ -87,8 +87,8 @@ def get_completion(query: str, model, tokenizer) -> str:
   generated_ids = model.generate(**model_inputs, max_new_tokens=1000, do_sample=True, pad_token_id=tokenizer.eos_token_id)
   decoded = tokenizer.batch_decode(generated_ids)
   return (decoded[0])
-  
-'''
+
+```
 
 # Load the Lora model
 model = PeftModel.from_pretrained(model, peft_model_id)