Update README.md

README.md

@@ -74,7 +74,7 @@ A Gemma-2b finetuned LoRA trained on science Q&A
 Users (both direct and downstream) should be made aware of the risks, biases and limitations of the model. More information needed for further recommendations.
 
 ## How to Get Started with the Model
-
+```
 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer, GenerationConfig
 from peft import PeftModel
@@ -122,6 +122,7 @@ with torch.no_grad():
 
 print(f"Inference time: {end-start:.2f} seconds")
 print(response)
+```
 [More Information Needed]
 
 ## Training Details