.

app.py

@@ -24,7 +24,7 @@ See how it works [here](http://fras.uk/ml/large%20prior-free%20models/transforme
 ''')
 
 st.markdown('''
-### t5-vae-python
+### [t5-vae-python](https://huggingface.co/flax-community/t5-vae-python)
 
 This model is trained on lines of Python code from GitHub ([dataset](https://huggingface.co/datasets/Fraser/python-lines)).
 ''')
@@ -120,11 +120,11 @@ st.write(out)
 
 
 st.markdown('''
-### t5-vae-wiki
+### [t5-vae-wiki](https://huggingface.co/flax-community/t5-vae-wiki)
 
 This model is trained on just 5% of the sentences on wikipedia.
 
-We'll release another model trained on the whole dataset soon.
+We'll release another model trained on the full [dataset](https://github.com/ChunyuanLI/Optimus/blob/master/download_datasets.md) soon.
 ''')