Update README.md

README.md

@@ -12,6 +12,15 @@ The aim of this work is to map a simple distribution - which is easy to sample a
 
 This model is trained on a toy dataset, the make_moons from sklearn.datasets.
 
+## Requirements
+
+This model requires
+- Tensorflow 2.3
+- Tensorflow probability 0.11.0
+
+## Loading this Model
+
+Only the model weights are stored. To load this model, use `load_model` function from load_model.py.
 ### Training hyperparameters
 
 The following hyperparameters were used during training: