Update README.md

README.md

@@ -65,6 +65,12 @@ Pragna-1B is trained on our proprietary platform, GenAI Studio, a modular AI Dev
 ## How to Get Started with the Model
 
 Use the code below to get started with the model.
+```python
+from transformers import AutoTokenizer, AutoModelForCausalLM
+tokenizer = AutoTokenizer.from_pretrained("soketlabs/pragna-1b")
+model = AutoModelForCausalLM.from_pretrained(
+    "soketlabs/pragna-1b", torch_dtype=torch.bfloat16
+)```
 
 [More Information Needed]