Update README.md

README.md

@@ -31,3 +31,9 @@ import torch
 
 model_path = hf_hub_download("username/medical-diffusion-classifier", "classification_model_256.pth.tar")
 model = torch.load(model_path, weights_only=False)
+
+## 🧾 Citation
+
+If you use this model, please cite our work:
+
+Cahan, N. et al. X-ray2CTPA: leveraging diffusion models to enhance pulmonary embolism classification. npj Digit. Med. 8, 439 (2025). https://doi.org/10.1038/s41746-025-01857-y