Update README.md

README.md

@@ -17,7 +17,11 @@ license: mit
     <img src="https://img.shields.io/badge/Discord-Join-5865F2?style=for-the-badge&logo=discord&logoColor=white" alt="Discord"/>
   </a>
   <a href="https://github.com/Lexsi-Labs/Orion-BiX">
-    <img src="https://img.shields.io/badge/GitHub-Repository-181717?style=for-the-badge&logo=github&logoColor=white" alt="GitHub"/>
+    <img src="https://img.shields.io/badge/GitHub-Orion%20BiX-181717?style=for-the-badge&logo=github&logoColor=white" alt="GitHub"/>
+  </a>
+  <!-- TabTune repo -->
+  <a href="https://github.com/Lexsi-Labs/TabTune">
+    <img src="https://img.shields.io/badge/GitHub-TabTune-181717?style=for-the-badge&logo=github&logoColor=white" alt="TabTune GitHub"/>
   </a>
 </div>
 
@@ -52,6 +56,8 @@ Input → tf_col (Set Transformer) → Bi-Axial Attention → tf_icl (ICL) → O
 
 
 ## Usage
+### Direct (OrionBiX Python package)
+
 
 ```python
 from orion_bix.sklearn import OrionBiXClassifier
@@ -64,9 +70,37 @@ predictions = clf.predict(X_test)
 
 This code will automatically download the pre-trained model from Hugging Face and use a GPU if available.
 
+### Via TabTune (unified TFM library)
+
+OrionBix can be used either directly via its own Python package or through [TabTune](https://github.com/Lexsi-Labs/TabTune), which provides a unified interface over several tabular foundation models.
+
+```python
+from tabtune import TabularPipeline
+
+pipeline = TabularPipeline(
+    model_name="OrionBix",          # use OrionBix through TabTune
+    tuning_strategy="inference",    # zero-shot / in-context mode
+    tuning_params={"device": "cuda"}  # or "cpu"
+)
+
+pipeline.fit(X_train, y_train)
+predictions = pipeline.predict(X_test)
+```
+
+When used through TabTune, the OrionBiX weights are automatically downloaded from this Hugging Face repository on first use, and TabTune handles model-aware preprocessing for you.
+
+
 ## Installation
 
-### From the source
+### Via TabTune (recommended if you want multiple tabular FMs)
+
+```bash
+pip install tabtune
+```
+
+This installs TabTune and its built-in OrionBiX integration; no separate orion-bix install is required.
+
+### From the OrionBiX source
 #### Option 1: From the local clone
 
 ```bash
@@ -82,12 +116,16 @@ pip install git+https://github.com/Lexsi-Labs/Orion-BiX.git
 
 ## Citation
 
-If you use Orion-BiX in your research, please cite:
+If you use Orion-BiX in your research, please cite our [paper](https://arxiv.org/abs/2512.00181)::
 
 ```bibtex
-@misc{bouadi25oriobix,
-  title={Orion-BiX: Bi-Axial Meta-Learning for Tabular In-Context Learning},
-  author={Mohamed Bouadi and Pratinav Seth and Aditya Tanna and Vinay Kumar Sankarapu},
-  year={2025},
+@misc{bouadi2025orionbix,
+      title={Orion-Bix: Bi-Axial Attention for Tabular In-Context Learning}, 
+      author={Mohamed Bouadi and Pratinav Seth and Aditya Tanna and Vinay Kumar Sankarapu},
+      year={2025},
+      eprint={2512.00181},
+      archivePrefix={arXiv},
+      primaryClass={cs.LG},
+      url={https://arxiv.org/abs/2512.00181}, 
 }
 ```