added to README how to save generated image
Browse files
README.md
CHANGED
@@ -11,12 +11,16 @@ tags:
|
|
11 |
- unconditional
|
12 |
- wsi
|
13 |
---
|
14 |
-
A Diffusion Model for Generating WSI
|
15 |
|
16 |
How to use the model?
|
17 |
|
18 |
```py
|
19 |
from diffusers import DiffusionPipeline
|
20 |
|
21 |
-
|
|
|
|
|
|
|
|
|
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 |
```
|