diff --git a/frontend/css/styles.css b/frontend/css/styles.css
index 5995710..46ee4aa 100644
--- a/frontend/css/styles.css
+++ b/frontend/css/styles.css
@@ -17,3 +17,8 @@ input[type=number]:disabled { -moz-appearance: textfield; }
 #generate, #img2img_mask_btn, #img2img_edit_btn {
     align-self: stretch;
 }
+
+#gan_image {
+    width: 512px;
+    height: auto;
+}
diff --git a/frontend/frontend.py b/frontend/frontend.py
index f11917c..0cb0b83 100644
--- a/frontend/frontend.py
+++ b/frontend/frontend.py
@@ -311,7 +311,7 @@ def draw_gradio_ui(opt, img2img=lambda x: x, txt2img=lambda x: x, txt2img_defaul
                                                         value=gfpgan_defaults['strength'])
                             gfpgan_btn = gr.Button("Generate", variant="primary")
                         with gr.Column():
-                            gfpgan_output = gr.Image(label="Output")
+                            gfpgan_output = gr.Image(label="Output", elem_id='gan_image')
                     gfpgan_btn.click(
                         run_GFPGAN,
                         [gfpgan_source, gfpgan_strength],
@@ -328,7 +328,7 @@ def draw_gradio_ui(opt, img2img=lambda x: x, txt2img=lambda x: x, txt2img_defaul
                                                                 value='RealESRGAN_x4plus')
                             realesrgan_btn = gr.Button("Generate")
                         with gr.Column():
-                            realesrgan_output = gr.Image(label="Output")
+                            realesrgan_output = gr.Image(label="Output", elem_id='gan_image')
                     realesrgan_btn.click(
                         run_RealESRGAN,
                         [realesrgan_source, realesrgan_model_name],