Update app.py

app.py

@@ -88,9 +88,9 @@ with gr.Blocks() as demo:
             btn_process = gr.Button(label="Process")
 
         with gr.Column():
-            # output_plot = gr.Plot(label="Segmentation Result")
+            output_plot = gr.Plot(label="Segmentation Result")
             output_mask = gr.Image(label="Mask")
-            # output_image = gr.Image(label="Output Image")
+            output_image = gr.Image(label="Output Image")
 
     btn_process.click(
         process_image,
@@ -98,10 +98,10 @@ with gr.Blocks() as demo:
             input_image,
             input_prompt,
             input_slider_T,
-            input_slider_A
+            input_slider_A,
+            draw_rectangles,
         ],
-        outputs=[output_mask],
-        api_name="masking"
+        outputs=[output_plot, output_mask, output_image],api_name="masking"
     )
 
     gr.Examples(
@@ -119,4 +119,4 @@ with gr.Blocks() as demo:
         ],
     )
 
-demo.launch()
+demo.launch()