Update README.md

README.md

@@ -55,11 +55,29 @@ More for DreamBooth model, please refer to  https://dreambooth.github.io/.
 ## Usage
 
 ```python
+pip install -qqU diffusers accelerate
+
+import torch
 from diffusers import StableDiffusionPipeline
 
-pipeline = StableDiffusionPipeline.from_pretrained('li-yan/stable-diffusion-crazy-keyboard')
-image = pipeline().images[0]
-image
+pipe = StableDiffusionPipeline.from_pretrained(model_id)
+
+# set prompt
+prompt = "a photo of cyberboard keyboard" #@param
+guidance_scale=12 #@param {type:"integer"}
+num_inference_steps = 50 #@param {type:"integer"}
+
+# Run the pipeline, showing some of the available arguments
+pipe_output = pipe(
+    prompt=prompt, # What to generate
+    negative_prompt="Oversaturated, blurry, low quality", # What NOT to generate
+    height=480, width=640,                                # Specify the image size
+    guidance_scale=guidance_scale,                        # How strongly to follow the prompt
+    num_inference_steps=num_inference_steps,              # How many steps to take
+)
+
+# View the resulting image
+pipe_output.images[0]
 ```
 
 ## Fine Tune Source Code