update README

README.md

@@ -30,36 +30,16 @@ Available models:
 
 ## Usage
 
-```python
-import torch
-from huggingface_hub import hf_hub_download
-from tryoffdiff.modeling.model import create_model
+> For a complete running example, see the demo code on [HuggingFace Spaces][hf_spaces].
 
-repo = "rizavelioglu/tryoffdiff"  # Define the repository
-
-# --- Helper function to load a model ---
-def load_tryoffdiff_model(model_name, filename, device="cuda"):
-    model = create_model(model_name)
-    model_path = hf_hub_download(repo_id=repo, filename=filename)
-    model.load_state_dict(torch.load(model_path, weights_only=True))
-    return model.eval().to(device)
-
-tryoffdiffv1       = load_tryoffdiff_model("TryOffDiff", "tryoffdiff.pth")
-tryoffdiffv2_upper = load_tryoffdiff_model("TryOffDiffv2_single", "tryoffdiffv2_upper.pth")
-tryoffdiffv2_lower = load_tryoffdiff_model("TryOffDiffv2_single", "tryoffdiffv2_lower.pth")
-tryoffdiffv2_dress = load_tryoffdiff_model("TryOffDiffv2_single", "tryoffdiffv2_dress.pth")
-tryoffdiffv2_multi = load_tryoffdiff_model("TryOffDiffv2", "tryoffdiffv2_multi.pth")
-```
-
-> For a complete running example, see the demo code on [HuggingFace Spaces][hf_spaces].\
 > For more information on training, inference, and evaluation, refer to the [GitHub repository][github].
 
-### License
+## 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
+## Citation
 If you use this work, please give a star ⭐ and a citation:
 ```
 @article{velioglu2024tryoffdiff,