Update app.py

app.py

@@ -89,28 +89,26 @@ with gr.Blocks(css=css) as demo:
   ![Examples](https://multidiffusion.github.io/pics/tight.jpg)
   ''')
   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.Box(elem_id="main-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()
+    out_image = gr.Image(label="Result")
   
   css_height = gr.HTML("<style>#main-image{width: 512px} .fixed-height{height: 512px !important}</style>")