update README

README.md

@@ -2,4 +2,66 @@
 license: other
 license_name: server-side-public-license
 license_link: https://www.mongodb.com/legal/licensing/server-side-public-license
+tags:
+- diffusion
+- virtual try-on
+- virtual try-off
+- image generation
+- fashion
+- e-commerce
+base_model:
+- CompVis/stable-diffusion-v1-4
+pipeline_tag: image-to-image
+library_name: diffusers
 ---
+
+## TryOffDiff
+
+The models proposed in the paper _"TryOffDiff: Virtual-Try-Off via High-Fidelity Garment Reconstruction\\using Diffusion Models"_
+[[paper]](https://arxiv.org/abs/2404.08582) [[project page]][project_page]:
+- `tryoffdiff.pth`: The pre-trained StableDiffusion-v1.4 fine-tuned on `VITON-HD-train` dataset.
+- `.pth`: A U-Net trained from scratch on `VITON-HD-train` dataset.
+- `.pth`: 
+
+
+## Usage
+
+```python
+from torchvision.io import read_image
+from torchvision.models.detection import MaskRCNN_ResNet50_FPN_Weights
+from huggingface_hub import hf_hub_download
+
+path_onnx = hf_hub_download(
+    repo_id="rizavelioglu/tryoffdiff",
+    filename="tryoffdiff.pth",  # or one of ["ldm-1", "ldm-2", "ldm-3", ...]
+)
+
+```
+
+> Check out the demo code on [HuggingFace Spaces][hf_spaces] for visualizing the output.
+
+> Also, check out [GitHub repository][project_page] to get more information on
+> training, inference, and evaluation.
+
+### License
+TL;DR: Not available for commercial use, unless the FULL source code is shared! \
+This project is intended solely for academic research. No commercial benefits are derived from it.
+Models are licensed under [Server Side Public License (SSPL)][license]
+
+### Citation
+If you find this repository useful in your research, please consider giving a star ⭐ and a citation:
+```
+@article{velioglu2024tryoffdiff,
+  title     = {TryOffDiff: Virtual-Try-Off via High-Fidelity Garment Reconstruction using Diffusion Models},
+  author    = {Velioglu, Riza and Bevandic, Petra and Chan, Robin and Hammer, Barbara},
+  journal   = {arXiv},
+  year      = {2024},
+  note      = {\url{https://doi.org/....}}
+}
+```
+
+[hf_spaces]: https://huggingface.co/spaces/rizavelioglu/tryoffdiff
+[project_page]: https://rizavelioglu.github.io/tryoffdiff/
+[github]: https://github.com/rizavelioglu/tryoffdiff
+[license]: https://www.mongodb.com/legal/licensing/server-side-public-license
+