Update app.py

app.py

@@ -133,7 +133,7 @@ with gr.Blocks(css=css) as demo:
             with gr.Row():
                 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=355)
+            output_image = gr.Image(label="Output", height=355, elem_id="output_image")
             with gr.Row(visible=False, elem_id="post_gen_info") as post_gen_info:
                 with gr.Column(min_width=10):
                     thumbs_up = gr.Button("👍")