Update README.md

README.md

@@ -63,14 +63,16 @@ To run inference with the Llama-3-SEC model using the llama3 chat template, use
 
 ```python
 from transformers import AutoModelForCausalLM, AutoTokenizer
-device = "cuda" 
+device = "cuda"
+
+model_name = "arcee-ai/Llama-3-SEC"
 
 model = AutoModelForCausalLM.from_pretrained(
-    "arcee-ai/Llama-3-SEC",
+    "model_name",
     torch_dtype="auto",
     device_map="auto"
 )
-tokenizer = AutoTokenizer.from_pretrained("Qwen/Qwen2-72B-Instruct")
+tokenizer = AutoTokenizer.from_pretrained("model_name")
 
 prompt = "What are the key regulatory considerations for a company planning to conduct an initial public offering (IPO) in the United States?"
 messages = [