Update README.md

README.md

@@ -35,18 +35,20 @@ You can also run this model using the following code:
 import transformers
 from transformers import AutoTokenizer
 
+
+model = "joey00072/ToxicHermes-2.5-Mistral-7B"
 # Format prompt
 message = [
     {"role": "system", "content": "You are a helpful assistant chatbot."},
     {"role": "user", "content": "What is a Large Language Model?"}
 ]
-tokenizer = AutoTokenizer.from_pretrained(new_model)
+tokenizer = AutoTokenizer.from_pretrained(model)
 prompt = tokenizer.apply_chat_template(message, add_generation_prompt=True, tokenize=False)
 
 # Create pipeline
 pipeline = transformers.pipeline(
     "text-generation",
-    model=new_model,
+    model=model,
     tokenizer=tokenizer
 )