Update README.md

README.md

@@ -73,14 +73,32 @@ Users (both direct and downstream) should be made aware of the risks, biases and
 
 Use the code below to get started with the model.
 ```python
-from transformers import pipeline
+from transformers import (
+    AutoModelForCausalLM,
+    AutoTokenizer,
+    BitsAndBytesConfig,
+    HfArgumentParser,
+    TrainingArguments,
+    pipeline,
+    logging,
+)
+
+
 model_id = "zephyr-7b-beta-Agent-Instruct-math"
+base_model = "HuggingFaceH4/zephyr-7b-beta"
+tokenizer = AutoTokenizer.from_pretrained(base_model, trust_remote_code=True)
+tokenizer.padding_side = 'right'
+tokenizer.pad_token = tokenizer.eos_token
+tokenizer.add_eos_token = True
+tokenizer.add_bos_token, tokenizer.add_eos_token
+
+
 pipe = pipeline(task="text-generation", model=model, tokenizer=tokenizer, max_length=600)
 prompt = "Enter your query"
 result = pipe(prompt)
 print(result[0]['generated_text'])
 
-
+```
 
 [More Information Needed]