hjjeon commited on
Commit
78ee1f0
1 Parent(s): 3fc128f

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +12 -1
README.md CHANGED
@@ -22,7 +22,18 @@ on the `huggan/smithsonian_butterflies_subset` dataset.
22
  #### How to use
23
 
24
  ```python
25
- # TODO: add an example code snippet for running this diffusion pipeline
 
 
 
 
 
 
 
 
 
 
 
26
  ```
27
 
28
  #### Limitations and bias
 
22
  #### How to use
23
 
24
  ```python
25
+ from diffusers import DDPMPipeline
26
+
27
+ model_id = "hjjeon/ddpm-butterflies-128"
28
+
29
+ # load model and scheduler
30
+ pipeline = DDPMPipeline.from_pretrained(model_id)
31
+
32
+ # run pipeline in inference
33
+ image = pipeline()["sample"]
34
+
35
+ # save image
36
+ image[0].save("butterfly.png")
37
  ```
38
 
39
  #### Limitations and bias