Update README.md

README.md

@@ -7,15 +7,22 @@ extra_gated_fields:
   Affiliation: text
   I want to use this model on the following data:
     type: select
-    options: 
-      - Legacy Survey
-      - HSC
-      - SDSS
-      - DESI
-      - Gaia
+    options:
+    - Legacy Survey
+    - HSC
+    - SDSS
+    - DESI
+    - Gaia
   I want to use this model for the following science case: text
   I agree to provide feedback on the model and API through GitHub issues during the alpha-testing phase: checkbox
   I agree to use this model for non-commercial use ONLY: checkbox
+license: mit
+datasets:
+- MultimodalUniverse/legacysurvey
+- MultimodalUniverse/hsc
+- MultimodalUniverse/gaia
+- MultimodalUniverse/sdss
+- MultimodalUniverse/desi
 ---
 # AION-1: Astronomical Omnimodal Network
 
@@ -57,13 +64,9 @@ AION-Base processes data from major astronomical surveys:
 ## Installation
 
 ```bash
-pip install aion
+pip install polymathic-aion
 ```
 
-For development:
-```bash
-pip install aion[torch,dev]
-```
 
 ## Resources
 
@@ -71,26 +74,9 @@ pip install aion[torch,dev]
 - Documentation: https://polymathic-ai.github.io/AION/
 - Tutorial: https://colab.research.google.com/github/PolymathicAI/AION/blob/main/notebooks/Tutorial.ipynb
 
-## Model Variants
-
-| Variant | Parameters | Encoder/Decoder | Model Dim |
-|---------|------------|-----------------|-----------|
-| Base    | 300M       | 12/12 blocks    | 768       |
-| Large   | 800M       | 24/24 blocks    | 1024      |
-| XLarge  | 3B         | 24/24 blocks    | 2048      |
-
 ## License
 
 MIT License - see https://github.com/PolymathicAI/AION/blob/main/LICENSE for details.
 
-## Citation
-
-@article{aion2025,
-  title={AION: Astronomical Omnimodal Network},
-  author={...},
-  journal={arXiv preprint arXiv:XXXX.XXXXX},
-  year={2025}
-}
-
 ---
-Built with ❤ for the astronomical community by https://polymathic-ai.org/
+Built with ❤ for the astronomical community by https://polymathic-ai.org/