import type { TaskDataCustom } from "../index.js";

const taskData: TaskDataCustom = {
	datasets: [
		{
			description: "Synthetic dataset, for image relighting",
			id: "VIDIT",
		},
		{
			description: "Multiple images of celebrities, used for facial expression translation",
			id: "huggan/CelebA-faces",
		},
		{
			description: "12M image-caption pairs.",
			id: "Spawning/PD12M",
		},
	],
	demo: {
		inputs: [
			{
				filename: "image-to-image-input.jpeg",
				type: "img",
			},
		],
		outputs: [
			{
				filename: "image-to-image-output.png",
				type: "img",
			},
		],
	},
	isPlaceholder: false,
	metrics: [
		{
			description:
				"Peak Signal to Noise Ratio (PSNR) is an approximation of the human perception, considering the ratio of the absolute intensity with respect to the variations. Measured in dB, a high value indicates a high fidelity.",
			id: "PSNR",
		},
		{
			description:
				"Structural Similarity Index (SSIM) is a perceptual metric which compares the luminance, contrast and structure of two images. The values of SSIM range between -1 and 1, and higher values indicate closer resemblance to the original image.",
			id: "SSIM",
		},
		{
			description:
				"Inception Score (IS) is an analysis of the labels predicted by an image classification model when presented with a sample of the generated images.",
			id: "IS",
		},
	],
	models: [
		{
			description: "An image-to-image model to improve image resolution.",
			id: "fal/AuraSR-v2",
		},
		{
			description: "Powerful image editing model.",
			id: "black-forest-labs/FLUX.1-Kontext-dev",
		},
		{
			description: "Virtual try-on model.",
			id: "yisol/IDM-VTON",
		},
		{
			description: "Image re-lighting model.",
			id: "kontext-community/relighting-kontext-dev-lora-v3",
		},
		{
			description: "Strong model for inpainting and outpainting.",
			id: "black-forest-labs/FLUX.1-Fill-dev",
		},
		{
			description: "Strong model for image editing using depth maps.",
			id: "black-forest-labs/FLUX.1-Depth-dev-lora",
		},
	],
	spaces: [
		{
			description: "Image editing application.",
			id: "black-forest-labs/FLUX.1-Kontext-Dev",
		},
		{
			description: "Image relighting application.",
			id: "lllyasviel/iclight-v2-vary",
		},
		{
			description: "An application for image upscaling.",
			id: "jasperai/Flux.1-dev-Controlnet-Upscaler",
		},
	],
	summary:
		"Image-to-image is the task of transforming an input image through a variety of possible manipulations and enhancements, such as super-resolution, image inpainting, colorization, and more.",
	widgetModels: ["Qwen/Qwen-Image"],
	youtubeId: "",
};

export default taskData;
