add info

app.py

@@ -3,7 +3,6 @@ import jax.numpy as jnp
 from transformers import AutoTokenizer
 from transformers.models.t5.modeling_flax_t5 import shift_tokens_right
 from t5_vae_flax_alt.src.t5_vae import FlaxT5VaeForAutoencoding
-import info
 
 
 st.set_page_config(
@@ -25,7 +24,9 @@ See how it works [here](http://fras.uk/ml/large%20prior-free%20models/transforme
 ''')
 
 st.text('''
-Try interpolating between lines of Python code using this T5-VAE.
+### t5-vae-python
+
+This model is trained on lines of Python code from GitHub ([dataset](https://huggingface.co/datasets/Fraser/python-lines).
 ''')
 
 
@@ -119,7 +120,11 @@ st.write(out)
 
 
 st.text('''
-Try interpolating between sentences from wikipedia using this T5-VAE.
+### t5-vae-wiki
+
+This model is trained on just 5% of the sentences on wikipedia.
+
+Hopefully we'll release another model trained on the whole dataset soon.
 ''')
 
 
@@ -146,6 +151,10 @@ st.write(out)
 
 st.text('''
 Try arithmetic in latent space.
+
+Here latent codes for each sentence are found and arithmetic is done with them.
+
+Here it runs the sum `C + (B - A) = ?`
 ''')
 
 

info.py

@@ -1,5 +0,0 @@
-
-BACKGROUND = """
-
-
-"""