update readme

README.md

@@ -6,10 +6,10 @@ tags:
 ---
 
 # Intro
-This is the model for our paper ["Melanoma Detection using Adversarial Training and Deep Transfer Learning"](https://arxiv.org/abs/2004.06824).
+This is the model for our paper ["Melanoma Detection using Adversarial Training and Deep Transfer Learning"](https://arxiv.org/abs/2004.06824). Code is available [here](https://github.com/hasibzunair/adversarial-lesions).
 
 ## Model description
-The model is trained on the ISIC 2016 Task 3 dataset. The architecture and algorithm is described in this [paper](https://arxiv.org/abs/2004.06824). Training details are [here](https://github.com/hasibzunair/adversarial-lesions).
+The model is trained on the ISIC 2016 Task 3 dataset. The architecture and algorithm is described in this [paper](https://arxiv.org/abs/2004.06824).
 
 ## Intended uses & limitations
 You can use the raw model for melanoma detection from skin lesion images.