Update app.py

app.py

@@ -73,11 +73,9 @@ with gr.Blocks(css=css) as demo:
         with gr.Column(scale=55):
             with gr.Group():
 
-                with gr.Row():
-                    generate = gr.Button(value="Generate").style(rounded=(False, True, True, False))
-                img = gr.Image(label="Source image", height=256, tool="editor", type="pil")
+                img = gr.Image(label="Input image", height=512, tool="editor", type="pil")
 
-                image_out = gr.Image(height=512)
+                image_out = gr.Image(label="Output image", height=512)
                 # gallery = gr.Gallery(
                 #     label="Generated images", show_label=False, elem_id="gallery"
                 # ).style(grid=[1], height="auto")
@@ -85,8 +83,8 @@ with gr.Blocks(css=css) as demo:
         with gr.Column(scale=45):
             with gr.Tab("Options"):
                 with gr.Group():
-                    source_prompt = gr.Textbox(label="Source prompt", placeholder="Source prompt describes the input image")
-                    target_prompt = gr.Textbox(label="Target prompt", placeholder="Target prompt describes the output image")
+                    source_prompt = gr.Textbox(label="Source prompt", placeholder="Source prompt describes the input image").style(rounded=(False, True, True, False))
+                    target_prompt = gr.Textbox(label="Target prompt", placeholder="Target prompt describes the output image").style(rounded=(True, False, False, True))
 
                     with gr.Row():
                         source_guidance_scale = gr.Slider(label="Source guidance scale", value=1, minimum=1, maximum=10)
@@ -100,7 +98,11 @@ with gr.Blocks(css=css) as demo:
                         width = gr.Slider(label="Width", value=512, minimum=64, maximum=1024, step=8)
                         height = gr.Slider(label="Height", value=512, minimum=64, maximum=1024, step=8)
 
-                    seed = gr.Slider(0, 2147483647, label='Seed', value=0, step=1)
+                    with gr.Row():
+                        seed = gr.Slider(0, 2147483647, label='Seed', value=0, step=1)
+
+                    with gr.Row():
+                        generate = gr.Button(value="Edit").style(rounded=(True, True, True, True))
 
     inputs = [source_prompt, target_prompt, source_guidance_scale, guidance_scale, num_inference_steps,
               width, height, seed, img, strength]