macadeliccc commited on
Commit
2663b39
·
1 Parent(s): 0e855b0

added an image save in text2image demo

Browse files

user cant see the generated image because it doesnt display anywhere and is destroyed after the python context is over

Files changed (1) hide show
  1. README.md +2 -0
README.md CHANGED
@@ -59,6 +59,8 @@ prompt = "A photograph of the inside of a subway train. There are raccoons sitti
59
 
60
  generator = torch.Generator(device="cpu").manual_seed(0)
61
  image = pipe(prompt, num_inference_steps=25, generator=generator).images[0]
 
 
62
  ```
63
 
64
  ### Image-2-Image
 
59
 
60
  generator = torch.Generator(device="cpu").manual_seed(0)
61
  image = pipe(prompt, num_inference_steps=25, generator=generator).images[0]
62
+
63
+ image.save("kandinsky.png")
64
  ```
65
 
66
  ### Image-2-Image