Update app.py

app.py

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