kaveh commited on
Commit
5fdc130
·
1 Parent(s): b6f6ff5

added to README how to save generated image

Browse files
Files changed (1) hide show
  1. README.md +6 -2
README.md CHANGED
@@ -11,12 +11,16 @@ tags:
11
  - unconditional
12
  - wsi
13
  ---
14
- A Diffusion Model for Generating WSI patches
15
 
16
  How to use the model?
17
 
18
  ```py
19
  from diffusers import DiffusionPipeline
20
 
21
- pipeline = DiffusionPipeline.from_pretrained("kaveh/wsi_generator")
 
 
 
 
22
  ```
 
11
  - unconditional
12
  - wsi
13
  ---
14
+ A Diffusion Model for Generating WSI Patches
15
 
16
  How to use the model?
17
 
18
  ```py
19
  from diffusers import DiffusionPipeline
20
 
21
+ wsi_generator = DiffusionPipeline.from_pretrained("kaveh/wsi_generator")
22
+ wsi_generator.to("cuda")
23
+
24
+ generated_image = wsi_generator().images[0]
25
+ generated_image.save("wsi_generated.png")
26
  ```