Update app.py

app.py

@@ -61,6 +61,7 @@ div#share-btn-container > div {flex-direction: row;background: black;align-items
 #share-btn-container div:nth-child(-n+2){width: auto !important;min-height: 0px !important;}
 #share-btn-container .wrap {display: none !important}
 #share-btn-container.hidden {display: none!important}
+#post_gen_info{margin-top: .5em}
 '''
 
 original_pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float16)
@@ -118,12 +119,12 @@ with gr.Blocks(css=css) as demo:
     with gr.Column(min_width=10, scale=16, elem_classes="plus_column"):
         with gr.Row():
             with gr.Column(min_width=10, scale=4, elem_classes="random_column"):
-              lora_1 = gr.Image(interactive=False, height=150, elem_classes="selected_random", show_share_button=False, show_download_button=False)
+              lora_1 = gr.Image(interactive=False, height=150, elem_classes="selected_random", elem_id="randomLoRA_1", show_share_button=False, show_download_button=False)
               lora_1_prompt = gr.Markdown(visible=False)
             with gr.Column(min_width=10, scale=1, elem_classes="plus_column"):
               plus = gr.HTML("+", elem_classes="plus_button")
             with gr.Column(min_width=10, scale=4, elem_classes="random_column"):
-              lora_2 = gr.Image(interactive=False, height=150, elem_classes="selected_random", show_share_button=False, show_download_button=False)
+              lora_2 = gr.Image(interactive=False, height=150, elem_classes="selected_random", elem_id="randomLoRA_2", show_share_button=False, show_download_button=False)
               lora_2_prompt = gr.Markdown(visible=False)
             with gr.Column(min_width=10, scale=2, elem_classes="plus_column"):
                equal = gr.HTML("=", elem_classes="plus_button")
@@ -133,7 +134,7 @@ with gr.Blocks(css=css) as demo:
                 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)
-            with gr.Row(visible=False) as post_gen_info:
+            with gr.Row(visible=False, elem_id="post_gen_info") as post_gen_info:
                 with gr.Column(min_width=10):
                     thumbs_up = gr.Button("👍")
                 with gr.Column(min_width=10):