carlosabadia commited on
Commit
feb3aff
1 Parent(s): 44f8391

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -1
README.md CHANGED
@@ -40,8 +40,10 @@ Model supported in a [Gradio](https://github.com/gradio-app/gradio) Web UI and C
40
  ## Usage
41
 
42
  ```python
 
43
  from diffusers import StableDiffusionPipeline, DPMSolverMultistepScheduler
44
 
 
45
  pipe = StableDiffusionPipeline.from_pretrained(
46
  "carlosabadia/hasbulla",
47
  scheduler = DPMSolverMultistepScheduler.from_pretrained("carlosabadia/hasbulla", subfolder="scheduler"),
@@ -52,7 +54,8 @@ guidance_scale = 7
52
  prompt = "A portrait of hasbulla person"
53
 
54
  images = pipe(prompt, num_images_per_prompt=1, num_inference_steps=50, guidance_scale=guidance_scale).images
55
- image[0]
 
56
  ```
57
 
58
  ![visitors](https://visitor-badge.glitch.me/badge?page_id=carlosabadia/hasbulla)
 
40
  ## Usage
41
 
42
  ```python
43
+ import torch
44
  from diffusers import StableDiffusionPipeline, DPMSolverMultistepScheduler
45
 
46
+
47
  pipe = StableDiffusionPipeline.from_pretrained(
48
  "carlosabadia/hasbulla",
49
  scheduler = DPMSolverMultistepScheduler.from_pretrained("carlosabadia/hasbulla", subfolder="scheduler"),
 
54
  prompt = "A portrait of hasbulla person"
55
 
56
  images = pipe(prompt, num_images_per_prompt=1, num_inference_steps=50, guidance_scale=guidance_scale).images
57
+ image = images[0]
58
+ image.save("hasbulla.png")
59
  ```
60
 
61
  ![visitors](https://visitor-badge.glitch.me/badge?page_id=carlosabadia/hasbulla)