Update README.md

README.md

@@ -1,3 +1,32 @@
----
-license: mit
----
+---
+license: mit
+---
+<div align="center">
+
+# Re-implementation of ControlNet with Shape Masks
+
+
+[[`GitHub`]](https://github.com/AlonzoLeeeooo/shape-guided-controlnet) / [[`Dataset`]]()
+</div>
+
+A re-implementation of ControlNet with shape masks.
+
+# Model Weights Structure
+```
+shape-guided-controlnet/
+└── annotators                        <----- Model weights of the shape mask annotator (`U2-Net`)
+    └── u2net.pth
+└── shape-guided-controlnet           <----- Model weights of the trained ControlNet with shape masks
+    ├── config.json
+    └── diffusion_pytorch_model.safetensors
+└── stable-diffusion-v1.5             <----- Model weights of Stable Diffusion v1.5
+    ├── feature_extractor
+    ├── scheduler
+    ├── text_encoder
+    ├── tokenizer
+    ├── unet
+    ├── vae
+    ├── model_index.json
+    ├── v1-5-pruned.safetensors
+    └── v1-inference.yaml
+```