Update README.md

README.md

@@ -41,19 +41,18 @@ Model supported in a [Gradio](https://github.com/gradio-app/gradio) Web UI and C
 
 ```python
 from diffusers import StableDiffusionPipeline, DPMSolverMultistepScheduler
-import torch
 
-model_id = "carlosabadia/hasbulla"
+pipe = StableDiffusionPipeline.from_pretrained(
+    "carlosabadia/hasbulla", 
+    scheduler = DPMSolverMultistepScheduler.from_pretrained("carlosabadia/hasbulla", subfolder="scheduler"),
+    torch_dtype=torch.float16,
+).to("cuda")
 
-# Use the DPMSolverMultistepScheduler (DPM-Solver++) scheduler here instead
-pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
-pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
-pipe = pipe.to("cuda")
+guidance_scale = 7
+prompt = "A portrait of hasbulla person"
 
-prompt = "a portrait of hasbulla person"
-image = pipe(prompt).images[0]
-    
-image.save("hasbulla.png")
+images = pipe(prompt, num_images_per_prompt=1, num_inference_steps=50, guidance_scale=guidance_scale).images
+image[0]
 ```
 
 ![visitors](https://visitor-badge.glitch.me/badge?page_id=carlosabadia/hasbulla)