Update app.py

app.py

@@ -29,16 +29,15 @@ def generate_image_by_description(description):
 
 
 with gr.Blocks() as demo:
-    selected_image = gr.Image(type="filepath", label="Upload an Image of the Pigeon")
-    generate_button = gr.Button("Generate Avatar", variant="primary")
-    generated_image = gr.Image(type="numpy", label="Generated Avatar")
-
-    # Function chaining: generate description, then generate image without displaying text
-    def process_and_generate(image):
-        description = generate_description(image)
-        return generate_image_by_description(description)
-
-    generate_button.click(process_and_generate, inputs=selected_image, outputs=generated_image)
-
-
+    with gr.Row():
+        with gr.Column(scale=2, min_width=300):
+            selected_image = gr.Image(type="filepath", label="Upload an Image of the Pigeon",height=300)
+            generate_button = gr.Button("Generate Avatar", variant="primary")
+        with gr.Column(scale=2, min_width=300):
+            generated_image = gr.Image(type="numpy", label="Generated Avatar", height=300)
+            def process_and_generate(image):
+                description = generate_description(image)
+                return generate_image_by_description(description)
+
+            generate_button.click(process_and_generate, inputs=selected_image, outputs=generated_image)
 demo.launch()