Spaces:

1aurent
/

ddpm-mnist

Sleeping

1aurent commited on Oct 6, 2023

Commit

6bc78c1

1 Parent(s): bb3e0ce

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -19,7 +19,7 @@ gr.Interface(
         gr.inputs.Slider(1, 100, label='Inference Steps', default=5, step=1),
         gr.inputs.Slider(0, 2147483647, label='Seed', default=random_seed, step=1),
     ],
-    outputs=gr.Image(shape=[128,128], type="pil", elem_id="output_image"),
     css="#output_image{width: 256px}",
     title="Unconditional butterflies",
     description="A DDPM scheduler and UNet model trained on the MNIST dataset for unconditional image generation.",

         gr.inputs.Slider(1, 100, label='Inference Steps', default=5, step=1),
         gr.inputs.Slider(0, 2147483647, label='Seed', default=random_seed, step=1),
     ],
+    outputs=gr.Image(shape=[28,28], type="pil", elem_id="output_image"),
     css="#output_image{width: 256px}",
     title="Unconditional butterflies",
     description="A DDPM scheduler and UNet model trained on the MNIST dataset for unconditional image generation.",