Update README.md

README.md

@@ -2,15 +2,16 @@
 
 Large language models (LLMs) such as generative pretrained transformers (GPTs) have shown potential for various commercial applications, but their applicability for materials design remains underexplored. In this work, AtomGPT is introduced as a model specifically developed for materials design based on transformer architectures, demonstrating capabilities for both atomistic property prediction and structure generation. This study shows that a combination of chemical and structural text descriptions can efficiently predict material properties with accuracy comparable to graph neural network models, including formation energies, electronic bandgaps from two different methods, and superconducting transition temperatures. Furthermore, AtomGPT can generate atomic structures for tasks such as designing new superconductors, with the predictions validated through density functional theory calculations. This work paves the way for leveraging LLMs in forward and inverse materials design, offering an efficient approach to the discovery and optimization of materials.
 
-## Forward model example
+
+
+## Forward model example (structure to property)
 
 python atomgpt/forward_models/forward_models.py --config_name atomgpt/examples/forward_model/config.json
 
-## Inverse model example
+## Inverse model example (property to structure)
 
 python atomgpt/inverse_models/inverse_models.py --config_name atomgpt/examples/inverse_model/config.json
 
-#python atomgpt/examples/inverse_model/run.py
 
 # Google colab/Jupyter notebook
 
@@ -20,3 +21,6 @@ python atomgpt/inverse_models/inverse_models.py --config_name atomgpt/examples/i
 
 
 (Documentation development is in progress...)
+
+![AtomGPT schematic](https://pubs.acs.org/cms/10.1021/acs.jpclett.4c01126/asset/images/large/jz4c01126_0001.jpeg)
+