Update README.md

README.md

@@ -74,6 +74,14 @@ Users (both direct and downstream) should be made aware of the risks, biases and
 
 Use the code below to get started with the model.
 
+```markdown
+Install the latest version of the following python libraries:
+-torch
+-accelerate
+-peft
+-bitsandbytes
+```
+
 ```python
 from transformers import AutoModelForCausalLM
 from peft import PeftModel
@@ -86,8 +94,8 @@ model = PeftModel.from_pretrained(base_model, peft_model_id)
 Run the model with a quantization configuration
 
 ```python
-import torch, accelerate, peft
-from transformers import AutoModelForCausalLM, BitsAndBytesConfig, pipeline
+import torch, accelerate, peft 
+from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig, pipeline
 from peft import PeftModel
 
 # Set up quantization configuration
@@ -110,6 +118,8 @@ base_model = AutoModelForCausalLM.from_pretrained(
 peft_model_id = "fc91/phi3-mini-instruct-full_ethics-lora"
 model = PeftModel.from_pretrained(base_model, peft_model_id)
 
+tokenizer = AutoTokenizer.from_pretrained("microsoft/Phi-3-mini-4k-instruct")
+
 messages = [ 
     {"role": "system", "content": "You are a helpful AI assistant that grounds all of its replies in ethical theories."}, 
     {"role": "user", "content": """I am driving a car, and I have to make a choice. A kid suddenly appears in the middle of the road chasing a ball. To save the kid, I