Upload folder using huggingface_hub

README.md

@@ -1,25 +1,47 @@
 ---
-            license: mit
-            tags:
-            - diffusion
-            - concept-erasure
-            - stable-diffusion
-            - esdx
-            - airliner
-            ---
-
-            # esdx_airliner
-
-            This is a concept-erased Stable Diffusion model using the **esdx** method to remove the concept **"airliner"**.
-
-            ## Usage
-
-            ```python
-            from diffusers import StableDiffusionPipeline
-            import torch
-
-            pipe = StableDiffusionPipeline.from_pretrained("ErasureResearch/esdx_airliner", torch_dtype=torch.float16).to("cuda")
-            prompt = "a photo of a airliner"
-            image = pipe(prompt).images[0]
-            image.show()
-            
+license: mit
+tags:
+- diffusion
+- concept-erasure
+- stable-diffusion
+- esdx
+- airliner
+datasets:
+- imagenet
+language:
+- en
+pipeline_tag: text-to-image
+---
+
+# esdx_airliner
+
+This is a concept-erased Stable Diffusion model using the **Exact Source Distillation (ESD-X)** method to remove the concept **"Airliner"**.
+
+## Method
+
+Exact Source Distillation (ESD-X) erases concepts by distilling knowledge while excluding specific concept representations.
+
+## Usage
+
+```python
+from diffusers import StableDiffusionPipeline
+import torch
+
+pipe = StableDiffusionPipeline.from_pretrained("ErasureResearch/esdx_airliner", torch_dtype=torch.float16).to("cuda")
+prompt = "a photo of a airliner"
+image = pipe(prompt).images[0]
+image.save("erased_airliner.png")
+```
+
+## Citation
+
+If you use this model in your research, please cite:
+
+```bibtex
+@article{concept_erasure_2024,
+  title={Concept Erasure in Diffusion Models},
+  author={ErasureResearch Team},
+  journal={Proceedings of...},
+  year={2024}
+}
+```