Update app.py

app.py

@@ -302,10 +302,19 @@ if __name__ == "__main__":
         gr.inputs.Dropdown(["merge", "first"], label="Inpaint Mode")
     ]
 
-    outputs = [
-        gr.outputs.Image(type="pil", label="Output Image"),
-        gr.outputs.Image(type="pil", label="Output Mask")
-    ]
+    outputs = []
+
+    if task_type == 'det':
+        outputs.append(gr.outputs.Image(type="pil", label="Output Image"))
+    elif task_type == 'seg' or task_type == 'automatic':
+        outputs.append(gr.outputs.Image(type="pil", label="Output Image"))
+        outputs.append(gr.outputs.Image(type="pil", label="Output Mask"))
+    elif task_type == 'inpainting':
+        outputs.append(gr.outputs.Image(type="pil", label="Output Image"))
+        outputs.append(gr.outputs.Image(type="pil", label="Output Mask"))
+    else:
+        print("task_type:{} error!".format(task_type))
+
 
     iface = gr.Interface(run_grounded_sam, inputs, outputs, title="Grounded SAM Demo")