Update README.md

README.md

@@ -35,10 +35,14 @@ This is the model card of a 🤗 transformers model that has been pushed on the
 
 ## Uses
 
+# load library 
+
 import torch
 from transformers import AutoTokenizer, AutoModelForCausalLM, BitsAndBytesConfig
 import re
 
+# load model
+
 model_id = "jaeyoungk/albatross" 
 bnb_config = BitsAndBytesConfig(
     load_in_4bit=True,
@@ -50,6 +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"""
@@ -95,6 +100,7 @@ def gen(x):
 
     return extracted_text
 
+# test the model
 gen('input your text here')