stevhliu HF staff commited on
Commit
5f91254
·
1 Parent(s): 62d82c2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -1,11 +1,12 @@
1
- from diffusers import DDPMPipeline
2
  import torch
3
  import PIL.Image
4
  import gradio as gr
5
  import random
6
  import numpy as np
7
 
8
- pipeline = DDPMPipeline.from_pretrained("johnowhitaker/ddpm-butterflies-32px")
 
9
 
10
  def predict(steps, seed):
11
  generator = torch.manual_seed(seed)
@@ -19,7 +20,7 @@ gr.Interface(
19
  gr.inputs.Slider(1, 100, label='Inference Steps', default=5, step=1),
20
  gr.inputs.Slider(0, 2147483647, label='Seed', default=random_seed, step=1),
21
  ],
22
- outputs=gr.Image(shape=[256,256], type="pil", elem_id="output_image"),
23
  css="#output_image{width: 256px}",
24
  title="Unconditional butterflies",
25
  description="A DDPM scheduler and UNet model trained on a subset of the <a href=\"https://huggingface.co/datasets/huggan/smithsonian_butterflies_subset\">Smithsonian Butterflies</a> dataset for unconditional image generation.",
 
1
+ from diffusers import DiffusionPipeline
2
  import torch
3
  import PIL.Image
4
  import gradio as gr
5
  import random
6
  import numpy as np
7
 
8
+ pipeline = DiffusionPipeline.from_pretrained("johnowhitaker/ddpm-butterflies-32px")
9
+ pipeline.to('cuda')
10
 
11
  def predict(steps, seed):
12
  generator = torch.manual_seed(seed)
 
20
  gr.inputs.Slider(1, 100, label='Inference Steps', default=5, step=1),
21
  gr.inputs.Slider(0, 2147483647, label='Seed', default=random_seed, step=1),
22
  ],
23
+ outputs=gr.Image(shape=[128,128], type="pil", elem_id="output_image"),
24
  css="#output_image{width: 256px}",
25
  title="Unconditional butterflies",
26
  description="A DDPM scheduler and UNet model trained on a subset of the <a href=\"https://huggingface.co/datasets/huggan/smithsonian_butterflies_subset\">Smithsonian Butterflies</a> dataset for unconditional image generation.",