Update README.md

README.md

@@ -52,6 +52,7 @@ The code below shows how to use `dlite-v1-124m` in the way which it was trained.
 ```python
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import numpy as np
+import re
 
 model_id = 'aisquared/dlite-v1-124m'
 
@@ -63,7 +64,7 @@ model = AutoModelForCausalLM.from_pretrained(model_id, trust_remote_code = True,
 ### Create the Prompt Format and Other Variables
 
 ```python
-PROMPT_FORMAT = """Below is an instruction that describes a task. Write a response that appropriately completes the request.
+PROMPT = """Below is an instruction that describes a task. Write a response that appropriately completes the request.
 
 ### Instruction:
 {instruction}