edit usage section

README.md

@@ -108,8 +108,19 @@ This model was created as part of the DreamBooth Hackathon 🔥. Visit the [orga
 
 ```python
 from diffusers import StableDiffusionPipeline
+import torch
+
+device = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')
+pipeline = StableDiffusionPipeline.from_pretrained('simonschoe/pokeball-machine').to(device)
+
+prompt = "a pkblz ball in the middle of a miniature jungle"
+
+image = pipeline(
+    prompt,
+    num_inference_steps=50,
+    guidance_scale=10,
+    num_images_per_prompt=1
+).images[0]
 
-pipeline = StableDiffusionPipeline.from_pretrained('simonschoe/pokeball-machine')
-image = pipeline().images[0]
 image
 ```