Update README.md

README.md

@@ -48,9 +48,10 @@ generation you should look at model like GPT2.
 
 ### How to use
 
-```
+The full codebase is available on [GitHub](https://github.com/BoHuangLab/CELL-E_2).
 
-Here is how to use this model to get the features of a given text in PyTorch:
+```
+Here is how to use this model to do sequence prediction:
 
 ```python
 configs = OmegaConf.load(configs/config.yaml);