Upload README.md with huggingface_hub
Browse files
README.md
CHANGED
|
@@ -7,14 +7,27 @@ tags:
|
|
| 7 |
- diffusion-models-class
|
| 8 |
---
|
| 9 |
|
| 10 |
-
This model is a diffusion model for unconditional image generation of churches ⛪️ finetuned on wikiart
|
|
|
|
|
|
|
|
|
|
| 11 |
|
| 12 |
## Usage
|
| 13 |
|
| 14 |
```python
|
| 15 |
from diffusers import DDPMPipeline
|
| 16 |
|
| 17 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 18 |
image = pipeline().images[0]
|
| 19 |
-
|
|
|
|
|
|
|
| 20 |
```
|
|
|
|
|
|
|
|
|
|
|
|
| 7 |
- diffusion-models-class
|
| 8 |
---
|
| 9 |
|
| 10 |
+
This model is a diffusion model for unconditional image generation of churches ⛪️ finetuned on wikiart 🎨.<br>
|
| 11 |
+
|
| 12 |
+
Pretrained model : google/ddpm-church-256<br>
|
| 13 |
+
Dataset : huggan/wikiart<br>
|
| 14 |
|
| 15 |
## Usage
|
| 16 |
|
| 17 |
```python
|
| 18 |
from diffusers import DDPMPipeline
|
| 19 |
|
| 20 |
+
model_id = 'CCMat/ddpm-church-finetune-wikiart'
|
| 21 |
+
|
| 22 |
+
# load model and scheduler
|
| 23 |
+
pipeline = DDPMPipeline.from_pretrained(model_id)
|
| 24 |
+
|
| 25 |
+
# run pipeline in inference (sample random noise and denoise)
|
| 26 |
image = pipeline().images[0]
|
| 27 |
+
|
| 28 |
+
# save image
|
| 29 |
+
image.save("ddpm_church_wikiart.png")
|
| 30 |
```
|
| 31 |
+
|
| 32 |
+
## Samples
|
| 33 |
+
|