Update README.md

README.md

@@ -22,6 +22,22 @@ Compare result:
 
 ![](image1.jpg)
 
+# Usage
+
+```
+import torch
+from diffusers import DiffusionPipeline, DPMSolverMultistepScheduler
+
+repo_id = "mrdabin/poison"
+pipe = DiffusionPipeline.from_pretrained(repo_id, torch_dtype=torch.float16, revision="fp16")
+pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
+pipe = pipe.to("cuda")
+
+prompt = "High quality photo of an astronaut riding a horse in space"
+image = pipe(prompt, num_inference_steps=25).images[0]
+image.save("astronaut.png")
+```
+
 # License
 
 This model is open access and available to all, with a CreativeML OpenRAIL-M license further specifying rights and usage. The CreativeML OpenRAIL License specifies: