22

app.py

@@ -166,7 +166,7 @@ def inference(
         control_context_scale=control_context_scale,
     ).images[0]
 
-    return output_image, seed
+    return output_image, seed, control_image
 
 
 def read_file(path: str) -> str:
@@ -261,6 +261,11 @@ with gr.Blocks() as demo:
                         interactive=False,
                         type="pil",
                     )
+                    control_image = gr.Image(
+                        label="Generated Control Image", 
+                        interactive=False,
+                        type="pil",
+                    )
                     
         gr.Examples(examples=examples, inputs=[edit_dict, prompt])
         gr.Markdown(read_file("static/footer.md"))
@@ -283,7 +288,7 @@ with gr.Blocks() as demo:
             guidance_scale,
             num_inference_steps,
         ],
-        outputs=[output_image, seed],
+        outputs=[output_image, seed, control_image],
     )
 
 if __name__ == "__main__":