Update app.py

app.py

@@ -22,7 +22,7 @@ gr.Interface(
         grc.Slider(1, 100, label='Inference Steps', value=12, step=1),
         grc.Slider(0, 2147483647, label='Seed', value=69420, step=1),
     ],
-    outputs=gr.Image(shape=[28,28], type="pil", elem_id="output_image"),
+    outputs=gr.Image(height=28, width=28, type="pil", elem_id="output_image"),
     css="#output_image{width: 256px}",
     title="Unconditional MNIST",
     description="A DDIM scheduler and UNet model trained on the MNIST dataset for unconditional image generation.",