22

app.py

@@ -177,10 +177,11 @@ with gr.Blocks(css=css) as demo:
                 run_button = gr.Button("Run", variant="primary")
             with gr.Column():
                 output_image = gr.Image(label="Generated image", show_label=False)
-                
+                polished_prompt = gr.Textbox(label="Polished prompt", interactive=False)
+
                 with gr.Accordion("Preprocessor output", open=False):
                     control_image = gr.Image(label="Control image", show_label=False)
-                    polished_prompt = gr.Textbox(label="Polished prompt", interactive=False)
+                    
 
         with gr.Accordion("Advanced Settings", open=False):
                 seed = gr.Slider(
@@ -248,7 +249,7 @@ with gr.Blocks(css=css) as demo:
             guidance_scale,
             num_inference_steps,
         ],
-        outputs=[output_image, seed],
+        outputs=[output_image, seed, control_image],
     )
     # gr.on(
     #     triggers=[run_button.click, prompt.submit],