Update app.py

app.py

@@ -22,7 +22,8 @@ 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(height=28, width=28, type="pil"),
+    outputs=gr.Image(height=28, width=28, type="pil", elem_id="output_image"),
+    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()