Update README.md

README.md

@@ -1,67 +1,67 @@
----
-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
-
+---
+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/nt3n}}
+}
+```
+
+[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
+