Update README.md

README.md

@@ -3,7 +3,9 @@ pipeline_tag: image-to-text
 ---
 # Image captioning model
 
-End-to-end Transformer based image captioning model, where both the encoder and decoder use standard pre-trained transformer architectures.  
+End-to-end Transformer based image captioning model, where both the encoder and decoder use standard pre-trained transformer architectures. 
+
+### Repository: https://github.com/yesidc/image-captioning
 
 ## Encoder
 The encoder uses the pre-trained Swin transformer (Liu et al., 2021) that is a general-purpose backbone for computer vision. It outperforms ViT, DeiT and ResNe(X)t models at tasks such as image classification, object detection and semantic segmentation. The fact that this model is not pre-trained to be a 'narrow expert'--- a model pre-trained to perform a specific task e.g., image classification --- makes it a good candidate for fine-tuning on a downstream task.