Update app.py

app.py

@@ -67,15 +67,16 @@ with gr.Blocks(
             format="jpeg",
             interactive=False
         )
-        with gr.Row(equal_height=True):
-            prompt = gr.Textbox(
-                label="Prompt",
-                show_label=False,
-                placeholder="Enter your prompt..",
-                max_lines=1,
-                container=False
-            )
-            btn = gr.Button("Generate", variant="primary")
+        with gr.Group():
+            with gr.Row(equal_height=True):
+                prompt = gr.Textbox(
+                    label="Prompt",
+                    show_label=False,
+                    placeholder="Enter your prompt..",
+                    max_lines=1,
+                    container=False
+                )
+                btn = gr.Button("Generate", variant="primary", scale=0)
 
     with gr.Tab("💡 Examples"):
         examples = gr.Examples(