Update README.md

README.md

@@ -111,4 +111,18 @@ input_ids, attention_mask = preprocess_input(tokenizer, system_prompt, initial_q
 predicted_approach, _ = predict_approach(router_model, input_ids, attention_mask, device)
 
 print(f"Router predicted approach: {predicted_approach}")
+```
+
+## Citation
+
+If you use this in your work, please cite:
+
+```bibtex
+@software{optillm,
+  title = {Optillm: Optimizing inference proxy for LLMs},
+  author = {Asankhaya Sharma},
+  year = {2024},
+  publisher = {GitHub},
+  url = {https://github.com/codelion/optillm}
+}
 ```