Update README.md

README.md

@@ -6,17 +6,40 @@ metrics:
 - mae
 ---
 
+<div align="center">
+  [![License](https://img.shields.io/badge/Licenses-CC_BY_4.0-green)](./LICENSE)
+  [![Contact](https://img.shields.io/badge/Contact-SE-AOPT-blue)](mailto:SE-AOPT-office@helmholtz-berlin.de)
+</div>
+
 # OptoGPT++
 
 Meet OptoGPT++ — an enhanced implementation of the OptoGPT, a decoder-only transformer that aims to solve inverse design of multi-layer thin film structures.
 
 **Key Enhancements**
 
-- Inclusion 
+- Inclusion of an **absorption** feature in the model ➕📈
+- Increased the maximum **wave length** to 2,000nm 💡
+- Longer training time for **better predictive performance** 🤯
+
+Feel free to check out this supporting material
+
+**OptoGPT++**: [https://github.com/jnitzz/OptoLlama]
+**OptoGPT**: [https://github.com/taigaoma1997/optogpt]
+**ArXiV**: 📝 [https://arxiv.org/abs/2304.10294]
 
 ## Usage
 
-## Hyperparameters
+### Install Dependencies
+
+```bash
+python -m pip install torch
+python -m pip install safetensors
+```
+
+### Load Model Checkpoint
+
+```python
+```
 
 ## Acknowledgements
 
@@ -24,6 +47,8 @@ This work is supported by the Helmholtz Association Initiative and Networking Fu
 
 ## Citations
 
+If you find our work helpful, please feel free to cite as following:
+
 ```
 @article{ma2024optogpt,
   title={OptoGPT: a foundation model for inverse design in optical multilayer thin film structures},
@@ -35,4 +60,6 @@ This work is supported by the Helmholtz Association Initiative and Networking Fu
   publisher={Opto-Electronic Advance},
   doi={10.29026/oea.2024.240062}
 }
-```
+```
+
+----