update

app.py

@@ -21,6 +21,7 @@ title = r"""
 
 description = r"""
 <b>Official 🤗 Gradio demo</b> for <a href='https://github.com/QY-H00/attention-interpolation-diffusion/tree/public' target='_blank'><b>PAID: (Prompt-guided) Attention Interpolation of Text-to-Image Diffusion</b></a>.<br>
+We strongly recommend to deploy this demo locally with GPU resources for better experience.
 How to use:<br>
 1. Input prompt 1, prompt 2 and negative prompt.
 2. For <b> Compositional Generation </b> Input the guidance prompt and choose the one you are satisfied!
@@ -362,10 +363,10 @@ with gr.Blocks(css="style.css") as demo:
         )
 
     with gr.Group():
-        with gr.Row(elem_classes="flex-grow"):
-            image_prompt1 = gr.Image(label="Image Prompt 1 (Optional)", interactive=True, height=236, width=235)
-            image_prompt2 = gr.Image(label="Image Prompt 2 (Optional)", interactive=True, height=236, width=235)
-        with gr.Row(elem_classes="flex-grow"):
+        with gr.Row():
+            image_prompt1 = gr.Image(label="Image Prompt 1 (Optional)", interactive=True)
+            image_prompt2 = gr.Image(label="Image Prompt 2 (Optional)", interactive=True)
+        with gr.Row():
             model_choice = gr.Dropdown(
                 ["RealVis-v4.0", "SD1.4-512", "SD1.5-512", "SD2.1-768", "AOM3", "SDXL-1024", "Playground-XL-v2", "Juggernaut-XL-v9"],
                 label="Model",