Update README.md

README.md

@@ -70,6 +70,7 @@ Users (both direct and downstream) should be made aware of the risks, biases and
 ## How to Get Started with the Model
 
 Use the code below to get started with the model.
+```
 model_id = "mani-a-i/gemma_prvlaw_1e"
 model =  AutoModelForCausalLM.from_pretrained(model_id,device_map="auto")
 tokenizer = AutoTokenizer.from_pretrained("google/gemma-2b-it")
@@ -91,6 +92,8 @@ def gen(prompt):
 
 out = gen("Hi")
 
+```
+
 [More Information Needed]
 
 ## Training Details