22

app.py

@@ -168,7 +168,7 @@ def inference(
         control_context_scale=control_context_scale,
     ).images[0]
 
-    return output_image, seed, input_image
+    return output_image, seed
 
 
 def read_file(path: str) -> str:
@@ -263,11 +263,11 @@ with gr.Blocks() as demo:
                         interactive=False,
                         type="pil",
                     )
-                    control_image = gr.Image(
-                        label="Generated Control Image", 
-                        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"))
@@ -290,7 +290,7 @@ with gr.Blocks() as demo:
             # guidance_scale,
             num_inference_steps,
         ],
-        outputs=[output_image, seed, control_image],
+        outputs=[output_image, seed],
     )
 
 if __name__ == "__main__":