config.json

{
    "model_type": "u2net",
    "architectures": [
        "U2NET"
    ],
    "task": "image-segmentation",
    "pipeline_class": "u2net_pipeline.U2NetPipeline",
    "model_file": "u2net.py",
    "weights_file": "u2net.pth"
}