Update README.md

README.md

@@ -23,13 +23,13 @@ should probably proofread and complete it, then remove this comment. -->
 
 These are LoRA adaption weights for stabilityai/stable-diffusion-xl-base-1.0. The weights were fine-tuned on the Shuhaib73/stablediffusion_celeba1k dataset. You can find some example images in the following. 
 
-![img_0](./image_10.png)
+<!-- ![img_0](./image_10.png)
 ![img_1](./image_11.png)
 ![img_2](./image_21.png)
 ![img_3](./image_31.png)
 
-
-LoRA for the text encoder was enabled: False.
+ -->
+<!-- LoRA for the text encoder was enabled: False. -->
 
 Special VAE used for training: madebyollin/sdxl-vae-fp16-fix.
 
@@ -39,7 +39,15 @@ Special VAE used for training: madebyollin/sdxl-vae-fp16-fix.
 #### How to use
 
 ```python
-# TODO: add an example code snippet for running this diffusion pipeline
+import torch
+from diffusers import DiffusionPipeline
+
+model_path = "Shuhaib73/stablediffusion_fld"
+
+trained_pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float16)
+trained_pipe.to("cuda")
+trained_pipe.load_lora_weights(model_path)
+
 ```
 
 #### Limitations and bias