Update README.md

README.md

@@ -11,6 +11,27 @@ The training is based on [DiffEngine](https://github.com/okotaku/diffengine), th
 
 I used [diffusers/dog-example](https://huggingface.co/datasets/diffusers/dog-example).
 
+# Inference
+
+```
+import torch
+from diffusers import DiffusionPipeline
+
+checkpoint = 'takuoko/small-sd-dreambooth-lora-dog'
+prompt = 'A photo of sks dog in a bucket'
+
+pipe = DiffusionPipeline.from_pretrained(
+    'segmind/small-sd', torch_dtype=torch.float16)
+pipe.to('cuda')
+pipe.load_lora_weights(checkpoint)
+
+image = pipe(
+    prompt,
+    num_inference_steps=50,
+).images[0]
+image.save('demo.png')
+```
+
 # Example result
 
 prompt = 'A photo of sks dog in a bucket'