Update app.py

app.py

@@ -20,16 +20,19 @@ with gr.Blocks() as demo:
     with gr.Tab("ControlNet on Canny Filter "):
         prompt_input_canny = gr.Textbox(label="Prompt")
         negative_prompt_canny = gr.Textbox(label="Negative Prompt")
-        canny_input = gr.Image(label="Image")
+        canny_input = gr.Image(label="Input Image")
+        canny_output = gr.Image(label="Output Image")
         submit_btn = gr.Button(value = "Submit")
         canny_inputs = [prompt_input_canny, negative_prompt_canny, canny_input]
-        submit_btn.click(fn=infer_canny, inputs=canny_inputs, outputs=["image"])
+        submit_btn.click(fn=infer_canny, inputs=canny_inputs, outputs=[seg_output])
+        
     with gr.Tab("ControlNet with Semantic Segmentation"):
         prompt_input_seg = gr.Textbox(label="Prompt")
         negative_prompt_seg = gr.Textbox(label="Negative Prompt")
         seg_input = gr.Image(label="Image")
+        seg_output = gr.Image(label="Output Image")
         submit_btn = gr.Button(value = "Submit")
         seg_inputs = [prompt_input_seg, negative_prompt_seg, seg_input]
-        submit_btn.click(fn=infer_segmentation, inputs=seg_inputs, outputs=["image"])
+        submit_btn.click(fn=infer_segmentation, inputs=seg_inputs, outputs=[seg_output])
 
 demo.launch()