UPD: adding link to github

README.md

@@ -8,7 +8,9 @@ tags:
 ---
 # StainFuser: Controlling Diffusion for Faster Neural Style Transfer in Multi-Gigapixel Histology Images
 
-Repo containing model weights from the [paper](https://arxiv.org/abs/2403.09302)
+Repo containing model weights from the [paper](https://arxiv.org/abs/2403.09302), Github code can be found [here](https://github.com/R-J96/stainFuser)
+
+![StainFuser](assets/workflow_pipeline.png)
 
 ### Organisation
 - checkpoint: StainFuser trained weights trained at 512x512 resolution with mixed magnification