Update ui/gradio_ui.py

ui/gradio_ui.py

@@ -12,16 +12,19 @@ def ui(controller):#get_first_result,get_second_result): #controller):
             with gr.Column():
                 improved_sketch_view = gr.Image(type="pil", label="Improved Sketch")
 
+        
+
+        #with gr.Row():
+            with gr.Column():
+                result=gr.Image(type="pil", label="Improved Sketch")
+                second_prompt = gr.Textbox(label="Prompt", lines=3)
+                second_negative_prompt = gr.Textbox(label="Negative prompt", lines=3, value="disfigured, extra digit, fewer digits, cropped, worst quality, low quality")
+                result_button = gr.Button(value="Paint It", variant="primary")
+
         improve_sketch.click(fn=controller.get_first_result,
                              inputs=[sketch, first_prompt, first_negative_prompt],
                              outputs=improved_sketch_view)
 
-        with gr.Row():
-            result=gr.Image(type="pil", label="Improved Sketch")
-            second_prompt = gr.Textbox(label="Prompt", lines=3)
-            second_negative_prompt = gr.Textbox(label="Negative prompt", lines=3, value="disfigured, extra digit, fewer digits, cropped, worst quality, low quality")
-            result_button = gr.Button(value="Paint It", variant="primary")
-
         result_button.click(fn=controller.get_second_result,
                              inputs=[improved_sketch_view, second_prompt, second_negative_prompt],
                              outputs=result)