Update README.md

README.md

@@ -102,12 +102,27 @@ For detailed usage refer to the [colab_notebook](https://colab.research.google.c
 
 
 
-### Installation
+### Model Use
 
 ```bash
 pip install transformers
+pip install accelerate
 ```
 
+```python
+from transformers import AutoModelForCausalLM, AutoTokenizer
+from accelerate import Accelerator
+model =AutoModelForCausalLM.from_pretrained("PipableAI/pip-code-bandit",torch_dtype=torch.bfloat16,device_map="auto")
+tokenizer = tokenizer = AutoTokenizer.from_pretrained("PipableAI/pip-code-bandit")
+
+inputs = tokenizer(prompt, return_tensors="pt").to("cuda")
+outputs = model.generate(**inputs, max_new_tokens=new_tokens,)
+out = (
+    tokenizer.decode(outputs[0][inputs.input_ids.shape[-1]:], skip_special_tokens=True)
+)
+```
+
+
 ### Prompt