Update app.py

app.py

@@ -304,16 +304,16 @@ if __name__ == "__main__":
 
     outputs = []
 
-    if task_type == 'det':
+    if inputs["task_type"] == 'det':
         outputs.append(gr.outputs.Image(type="pil", label="Output Image"))
-    elif task_type == 'seg' or task_type == 'automatic':
+    elif inputs["task_type"] in ['seg', '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':
+    elif inputs["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))
+        print("task_type:{} error!".format(inputs["task_type"]))
 
 
     iface = gr.Interface(run_grounded_sam, inputs, outputs, title="Grounded SAM Demo")