Update README.md

README.md

@@ -21,15 +21,12 @@ should probably proofread and complete it, then remove this comment. -->
 
 # SDXL LoRA DreamBooth - reemas-irasna/home-decor_LoRA
 
-<Gallery
-
 ![image/png](https://cdn-uploads.huggingface.co/production/uploads/6638c2a3d5d215405e213a9b/6wr1qyRtCcdeMTVNqJghx.png)
 
 ![image/png](https://cdn-uploads.huggingface.co/production/uploads/6638c2a3d5d215405e213a9b/4xhhIJaZAFvmyls9Jb9wv.png)
 
 ![image/png](https://cdn-uploads.huggingface.co/production/uploads/6638c2a3d5d215405e213a9b/vY7jxRK7YbRaenJhHN4bs.png)
 
- />
 
 ## Model description
 
@@ -43,7 +40,7 @@ Special VAE used for training: madebyollin/sdxl-vae-fp16-fix.
 
 ## Trigger words
 
-You should use a photo of home to trigger the image generation.
+You should use "a photo of home" to trigger the image generation.
 
 ## Download model
 
@@ -58,13 +55,22 @@ Weights for this model are available in Safetensors format.
 #### How to use
 
 ```python
-# TODO: add an example code snippet for running this diffusion pipeline
+import torch
+from diffusers import DiffusionPipeline, AutoencoderKL
+
+vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
+pipe = DiffusionPipeline.from_pretrained(
+    "stabilityai/stable-diffusion-xl-base-1.0",
+    vae=vae,
+    torch_dtype=torch.float16,
+    variant="fp16",
+    use_safetensors=True
+)
+pipe.load_lora_weights("reemas-irasna/home-decor_LoRA")
+_ = pipe.to("cuda")
+
+prompt = "a photo of bedroom in red and white combination"
+
+image = pipe(prompt=prompt, num_inference_steps=25).images[0]
+image
 ```
-
-#### Limitations and bias
-
-[TODO: provide examples of latent issues and potential remediations]
-
-## Training details
-
-[TODO: describe the data used to train the model]