Update app.py

app.py

@@ -22,8 +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(height=28, width=28, type="pil", elem_id="output_image"),
-    css="#output_image{width: 256px !important; height: 256px !important;}",
+    outputs=gr.Image(height=28, width=28, type="pil"),
     title="Unconditional MNIST",
     description="A DDIM scheduler and UNet model trained on the MNIST dataset for unconditional image generation.",
 ).queue().launch()