Update README.md

Author: Abdalrahman112

README.md

@@ -7,25 +7,37 @@ The task of image captoning is the task of generating a sequence of words to des
 It is done through the well known sequence-to-sequence architecture.
 An encoder (of a mostly pretrained CNN model) encodes the image, then an RNN decoder outputs a word in each of its steps.
 
+I found this google documentation quite useful when coding both variants:
+
+[Google doc link](https://www.tensorflow.org/tutorials/text/image_captioning)
+
+
 
 # Usage
 You can either run image_captioning_train.ipynb for training the whole thing from the start while having the option of changing the hyper parameters or image_captioing.ipynb for captioing any image by giving its path.
 In case you use the latter, make sure you have 'Encoder.hdf5','Decoder.hdf5' and 'tokenizer.pickle' downloaded. 
 the encoder and decoder can be found in release.
+For the model with attention get the ones tagged with v2.0, otherwise get the ones tagged with v1.0.
 
 
 # Architecture
-This is a seq2seq neural network that uses teacher-forcing.
-This model doesn't use attention mechanism, which i might add later in another model
+There are two variants of the model that both use seq2seq architecture with teacher-forcing.
+Difference is that one uses attention mechanism while the other doesn't.
+You should go for the model variant with attention in almost all cases of usage as it is faster, smaller and more accurate.
+The other variant can be treated as a tutorial for those of you seeking basic understanding of the implementation of an image captioning model. This simpler archeticure is then built upon by the variant with attention.
 Encoding is done through using inception_v3.
 Feel free to change any of the parameters found in the configuration cell like LSTM_size, encoding_size or even the number of samples this network trains on.
 
 # Google colab links
 You can get the notebooks uploaded here via google colab through these links:
 
-[Training notebook](https://colab.research.google.com/drive/1VRQO7_r18a5rK68huvwKgNQOEGlzz3lm?usp=sharing)
+[Without attention training notebook](https://colab.research.google.com/drive/1VRQO7_r18a5rK68huvwKgNQOEGlzz3lm?usp=sharing)
+
+[Without attention prediction notebook](https://colab.research.google.com/drive/1OQEMKfT5BrTJpOwu__2u_tmNs8ADeQ6v?usp=sharing)
+
+[With attention prediction notebook](https://colab.research.google.com/drive/1Lz1vUFu2sx2AYWHJLSMwiG0V_ztdqEV2?usp=sharing)
 
-[Prediction notebook](https://colab.research.google.com/drive/1OQEMKfT5BrTJpOwu__2u_tmNs8ADeQ6v?usp=sharing)
+[With attention prediction notebook](https://colab.research.google.com/drive/1KcpU8vEw7Sozu-SBGAgULaNgAFZYDzkh?usp=sharing)
 
 # Example predictions
 ![Prediction1](https://scontent.fcai1-2.fna.fbcdn.net/v/t1.15752-9/96286228_934101440372416_1710498101753544704_n.png?_nc_cat=100&_nc_sid=b96e70&_nc_ohc=v68QTdYdXxQAX_N0Mrt&_nc_ht=scontent.fcai1-2.fna&oh=d10c6903515d3e2539f903b2b1d4422d&oe=5EDD55AE)