Update app.py

app.py

@@ -88,29 +88,29 @@ with gr.Blocks(css=css) as demo:
   This Space demonstrates MultiDiffusion region-based generation using Stable Diffusion model. To get started, draw your masks and type your prompts. More details in the [project page](https://multidiffusion.github.io).
   ![Examples](https://multidiffusion.github.io/pics/tight.jpg)
   ''')
-  with gr.Box(elem_id="main-image"):
-    #with gr.Accordion(open=True, label="Your color sketch") as sketch_accordion:  
-    with gr.Row():
+  with gr.Row():
+      with gr.Box(elem_id="main-image"):
+        #with gr.Accordion(open=True, label="Your color sketch") as sketch_accordion:  
         with gr.Column():
-            with gr.Row():
-                image = gr.Image(interactive=True, tool="color-sketch", source="canvas", type="pil")
-            with gr.Row():
-                aspect = gr.Radio(["Square", "Horizontal", "Vertical"], value="Square", label="Aspect Ratio")
-            button_run = gr.Button("I've finished my sketch",elem_id="main_button")
-            
-            prompts = []
-            colors = []
-            color_row = [None] * MAX_COLORS
-            with gr.Column(visible=False) as post_sketch:
-              general_prompt = gr.Textbox(label="General Prompt")
-              for n in range(MAX_COLORS):
-                with gr.Row(visible=False) as color_row[n]:
-                  with gr.Box(elem_id="color-bg"):
-                    colors.append(gr.HTML('<div class="color-bg-item" style="background-color: black"></div>'))
-                  prompts.append(gr.Textbox(label="Prompt for this color"))
-              final_run_btn = gr.Button("Generate!")
-        with gr.Column():
-            out_image = gr.Image()
+                with gr.Row():
+                    image = gr.Image(interactive=True, tool="color-sketch", source="canvas", type="pil")
+                with gr.Row():
+                    aspect = gr.Radio(["Square", "Horizontal", "Vertical"], value="Square", label="Aspect Ratio")
+                button_run = gr.Button("I've finished my sketch",elem_id="main_button")
+                
+                prompts = []
+                colors = []
+                color_row = [None] * MAX_COLORS
+                with gr.Column(visible=False) as post_sketch:
+                  general_prompt = gr.Textbox(label="General Prompt")
+                  for n in range(MAX_COLORS):
+                    with gr.Row(visible=False) as color_row[n]:
+                      with gr.Box(elem_id="color-bg"):
+                        colors.append(gr.HTML('<div class="color-bg-item" style="background-color: black"></div>'))
+                      prompts.append(gr.Textbox(label="Prompt for this color"))
+                  final_run_btn = gr.Button("Generate!")
+    
+        out_image = gr.Image()
   
   css_height = gr.HTML("<style>#main-image{width: 512px} .fixed-height{height: 512px !important}</style>")