Improve model card with metadata and GitHub link

This PR improves the model card by adding relevant metadata, including the pipeline tag and license. It also adds a link to the associated paper and GitHub repository.

README.md

@@ -1,7 +1,13 @@
+---
+pipeline_tag: image-to-image
+license: mit
+---
+
 # Medical Image VAE Models (Brain MRI)
 
 This repository contains Variational Autoencoder (VAE) models specifically adapted and fine-tuned for medical imaging tasks, particularly for brain MRI. Two different VAE model with different compression factor are provided (klf8, and klf4). Models are trained with KL loss, LIPIPs, and MSE loss.
 
+This model is used as part of the MAD-AD model, described in the paper [MAD-AD: Masked Diffusion for Unsupervised Brain Anomaly Detection](https://huggingface.co/papers/2502.16943). The code for the MAD-AD model can be found at https://github.com/farzad-bz/MAD-AD.
 
 ## Original Model Source
 
@@ -42,9 +48,4 @@ If you use this model in your research, please cite this work.
   journal={arXiv preprint arXiv:2502.16943},
   year={2025}
 }
-```
-
-
----
-license: mit
----
+```