update css to limit the output image dispaly size

app.py

@@ -14,5 +14,7 @@ def image_gen(name):
 	images = pipeline(num_inference_steps=40).images
 	return images[0]
 
-demo = gr.Interface(fn=image_gen, inputs=None, outputs="image")
+css = ".output-image, .input-image, .image-preview {height: 256px !important; width: 256px !important;}"
+
+demo = gr.Interface(fn=image_gen, inputs=None, outputs="image", css=css)
 demo.launch()