Update README.md

README.md

@@ -14,12 +14,25 @@ inference: true
     
 # LoRA text2image fine-tuning - johnowhitaker/lora_pn03_036sim
 
-These are LoRA adaption weights for playgroundai/playground-v2-1024px-aesthetic. The weights were fine-tuned on a custom dataset. You can find some example images in the following. 
-
+These are LoRA adaption weights for playgroundai/playground-v2-1024px-aesthetic. The weights were fine-tuned on a custom dataset.
 ![img_0](./image_0.png)
 ![img_1](./image_1.png)
-![img_2](./image_2.png)
-![img_3](./image_3.png)
+
+Usage:
+```python
+import torch
+from diffusers import DiffusionPipeline
+device = "cuda:7"
+pipe = DiffusionPipeline.from_pretrained(
+    'playgroundai/playground-v2-1024px-aesthetic',
+    torch_dtype=torch.float16,
+    add_watermarker=False,
+    variant="fp16"
+)
+pipe.to(device)
+pipe.load_lora_weights('johnowhitaker/lora_pn03_036sim')
+pipe("An owl wearing a tophat and tie, studio lighting, black background").images[0]
+```
 
 
 LoRA for the text encoder was enabled: False.