Update app.py

app.py

@@ -208,7 +208,7 @@ with gr.Blocks(css=css) as demo:
             with gr.Row(elem_id="prompt_area"):
                 prompt = gr.Textbox(label="Your prompt", info="Rearrange the trigger words into a coherent prompt", show_label=False, interactive=True, elem_id="prompt")
                 run_btn = gr.Button("Run", elem_id="run_button")
-            output_image = gr.Image(label="Output", height=460, elem_id="output_image", interactive=False)
+            output_image = gr.Image(label="Output", elem_id="output_image", interactive=False)
             with gr.Column(visible=False, elem_id="post_gen_info") as post_gen_info:
                 with gr.Row():
                     with gr.Column(min_width=10):