UPD: adding link to paper

README.md

@@ -8,11 +8,19 @@ tags:
 ---
 # StainFuser: Controlling Diffusion for Faster Neural Style Transfer in Multi-Gigapixel Histology Images
 
-Weights from paper (coming soon)
+Repo containing model weights from the [paper](https://arxiv.org/abs/2403.09302)
 
 ### Organisation
 - checkpoint: StainFuser trained weights trained at 512x512 resolution with mixed magnification
 - training: contains SD per-trained weights for backbone initialistaion in training
 
 ### Citation
-TBD
+@misc{jewsbury2024stainfuser,
+      title={StainFuser: Controlling Diffusion for Faster Neural Style Transfer in Multi-Gigapixel Histology Images}, 
+      author={Robert Jewsbury and Ruoyu Wang and Abhir Bhalerao and Nasir Rajpoot and Quoc Dang Vu},
+      year={2024},
+      eprint={2403.09302},
+      archivePrefix={arXiv},
+      primaryClass={eess.IV}
+}
+```