const taskData = {
    datasets: [
        {
            // TODO write proper description
            description: "Benchmark dataset used for image classification with images that belong to 100 classes.",
            id: "cifar100",
        },
        {
            // TODO write proper description
            description: "Dataset consisting of images of garments.",
            id: "fashion_mnist",
        },
    ],
    demo: {
        inputs: [
            {
                filename: "image-classification-input.jpeg",
                type: "img",
            },
        ],
        outputs: [
            {
                type: "chart",
                data: [
                    {
                        label: "Egyptian cat",
                        score: 0.514,
                    },
                    {
                        label: "Tabby cat",
                        score: 0.193,
                    },
                    {
                        label: "Tiger cat",
                        score: 0.068,
                    },
                ],
            },
        ],
    },
    metrics: [
        {
            description: "",
            id: "accuracy",
        },
        {
            description: "",
            id: "recall",
        },
        {
            description: "",
            id: "precision",
        },
        {
            description: "",
            id: "f1",
        },
    ],
    models: [
        {
            description: "A strong image classification model.",
            id: "google/vit-base-patch16-224",
        },
        {
            description: "A robust image classification model.",
            id: "facebook/deit-base-distilled-patch16-224",
        },
        {
            description: "A strong image classification model.",
            id: "facebook/convnext-large-224",
        },
    ],
    spaces: [
        {
            description: "A leaderboard to evaluate different image classification models.",
            id: "timm/leaderboard",
        },
    ],
    summary: "Image classification is the task of assigning a label or class to an entire image. Images are expected to have only one class for each image. Image classification models take an image as input and return a prediction about which class the image belongs to.",
    widgetModels: ["google/vit-base-patch16-224"],
    youtubeId: "tjAIM7BOYhw",
};
export default taskData;
