cap image size, ensure result dimension is divisible by 32

app.py

@@ -71,7 +71,7 @@ def process(input_image_editor, input_text, progress=gr.Progress(track_tqdm=True
         width=width,
         height=height,
         strength=0.7,
-        num_inference_steps=2
+        num_inference_steps=24
     ).images[0]
 
 
@@ -86,15 +86,16 @@ with gr.Blocks() as demo:
                 image_mode='RGB',
                 layers=False,
                 brush=gr.Brush(colors=["#FFFFFF"], color_mode="fixed"))
-            input_text_component = gr.Text(
-                label="Prompt",
-                show_label=False,
-                max_lines=1,
-                placeholder="Enter your prompt",
-                container=False,
-            )
-            submit_button_component = gr.Button(
-                value='Submit', variant='primary')
+            with gr.Row():
+                input_text_component = gr.Text(
+                    label="Prompt",
+                    show_label=False,
+                    max_lines=1,
+                    placeholder="Enter your prompt",
+                    container=False,
+                )
+                submit_button_component = gr.Button(
+                    value='Submit', variant='primary', scale=0)
         with gr.Column():
             output_image_component = gr.Image(
                 type='pil', image_mode='RGB', label='Generated image')