Update README.md

README.md

@@ -6,13 +6,11 @@ tags:
 - image-to-image
 - cyclegan
 - hed-to-anything
-base_model:
-- junyanz/pytorch-CycleGAN-and-pix2pix
 ---
 
 # CycleGAN_Hed2CoralReef Model
 
-This model transforms HED maps into coral reef style images, and also transforms coral reef style images into estimated HED maps using CycleGAN architecture.
+This model transforms HED maps into coral reef style images, and also transforms coral reef style images into estimated HED maps using the CycleGAN architecture via the junyanz's [pytorch-CycleGAN-and-pix2pix](https://github.com/junyanz/pytorch-CycleGAN-and-pix2pix).
 
 <div style="display: flex; flex-wrap: wrap; justify-content: center;">
   <div style="display: flex; width: 100%; justify-content: center; margin-bottom: 10px;">