Update README.md

README.md

@@ -35,13 +35,13 @@ This is the model card of a 🤗 transformers model that has been pushed on the
 
 ## Uses
 
-# load library 
+``` python 
 
 import torch
 from transformers import AutoTokenizer, AutoModelForCausalLM, BitsAndBytesConfig
 import re
 
-# load model
+
 
 model_id = "jaeyoungk/albatross" 
 bnb_config = BitsAndBytesConfig(
@@ -54,7 +54,7 @@ bnb_config = BitsAndBytesConfig(
 tokenizer = AutoTokenizer.from_pretrained('meta-llama/Meta-Llama-3-8B-Instruct')
 model = AutoModelForCausalLM.from_pretrained(model_id, quantization_config=bnb_config, device_map='auto')
 
-# define model generation function
+
 
 def gen(x):
     system_prompt = f"""