Add image-to-image and image-generation tasks

.gitignore

@@ -14,7 +14,6 @@ dist/
 downloads/
 eggs/
 .eggs/
-lib/
 lib64/
 parts/
 sdist/

js/src/lib/components/Icons/IconImageToImage.svelte

@@ -0,0 +1,25 @@
+<script lang="ts">
+	export let classNames = "";
+</script>
+
+<svg
+	class={classNames}
+	xmlns="http://www.w3.org/2000/svg"
+	xmlns:xlink="http://www.w3.org/1999/xlink"
+	aria-hidden="true"
+	fill="currentColor"
+	focusable="false"
+	role="img"
+	width="1em"
+	height="1em"
+	preserveAspectRatio="xMidYMid meet"
+	viewBox="0 0 32 32"
+>
+	<path
+		fill="currentColor"
+		d="M4 22H2V4a2.002 2.002 0 0 1 2-2h18v2H4zm17-5a3 3 0 1 0-3-3a3.003 3.003 0 0 0 3 3zm0-4a1 1 0 1 1-1 1a1 1 0 0 1 1-1z"
+	/><path
+		fill="currentColor"
+		d="M28 7H9a2.002 2.002 0 0 0-2 2v19a2.002 2.002 0 0 0 2 2h19a2.002 2.002 0 0 0 2-2V9a2.002 2.002 0 0 0-2-2Zm0 21H9v-6l4-3.997l5.586 5.586a2 2 0 0 0 2.828 0L23 22.003L28 27Zm0-3.828l-3.586-3.586a2 2 0 0 0-2.828 0L20 22.172l-5.586-5.586a2 2 0 0 0-2.828 0L9 19.172V9h19Z"
+	/>
+</svg>

js/src/lib/components/Icons/IconUnconditionalImageGeneration.svelte

@@ -0,0 +1,25 @@
+<script lang="ts">
+	export let classNames = "";
+</script>
+
+<svg
+	class={classNames}
+	xmlns="http://www.w3.org/2000/svg"
+	xmlns:xlink="http://www.w3.org/1999/xlink"
+	aria-hidden="true"
+	fill="currentColor"
+	focusable="false"
+	role="img"
+	width="1em"
+	height="1em"
+	preserveAspectRatio="xMidYMid meet"
+	viewBox="0 0 32 32"
+>
+	<path
+		fill="currentColor"
+		d="M19 14a3 3 0 1 0-3-3a3 3 0 0 0 3 3Zm0-4a1 1 0 1 1-1 1a1 1 0 0 1 1-1Z"
+	/><path
+		fill="currentColor"
+		d="M26 4H6a2 2 0 0 0-2 2v20a2 2 0 0 0 2 2h20a2 2 0 0 0 2-2V6a2 2 0 0 0-2-2Zm0 22H6v-6l5-5l5.59 5.59a2 2 0 0 0 2.82 0L21 19l5 5Zm0-4.83l-3.59-3.59a2 2 0 0 0-2.82 0L18 19.17l-5.59-5.59a2 2 0 0 0-2.82 0L6 17.17V6h20Z"
+	/>
+</svg>

js/src/lib/components/ModelPipelineIcon/ModelPipelineIcon.svelte

@@ -27,6 +27,8 @@
 	import IconVoiceActivityDetection from "../Icons/IconVoiceActivityDetection.svelte";
 	import IconZeroShotClassification from "../Icons/IconZeroShotClassification.svelte";
 	import IconReinforcementLearning from "../Icons/IconReinforcementLearning.svelte";
+	import IconImageToImage from "../Icons/IconImageToImage.svelte";
+	import IconUnconditionalImageGeneration from "../Icons/IconUnconditionalImageGeneration.svelte";
 
 	export let classNames = "";
 	export let pipeline = "";
@@ -61,6 +63,8 @@
 		"structured-data-classification": IconStructuredDataClassification,
 		"audio-classification": IconAudioClassification,
 		"reinforcement-learning": IconReinforcementLearning,
+		"image-to-image": IconImageToImage,
+		"unconditional-image-generation": IconUnconditionalImageGeneration,
 	};
 </script>
 

js/src/lib/interfaces/Types.ts

@@ -36,6 +36,8 @@ export enum PipelineType {
 	"image-segmentation"                                      = "Image Segmentation",
 	"text-to-image"                                           = "Text-to-Image",
 	"image-to-text"                                           = "Image-to-Text",
+	"image-to-image"                                          = "Image-to-Image",
+	"unconditional-image-generation"                          = "Unconditional Image Generation",
 	/// others
 	"structured-data-classification"                          = "Structured Data Classification",
 	"reinforcement-learning"                                  = "Reinforcement Learning",
@@ -100,6 +102,8 @@ export const PIPELINE_TAG_MODALITIES: Record<keyof typeof PipelineType, Modality
 	"image-segmentation":             "cv",
 	"text-to-image":                  "cv",
 	"image-to-text":                  "cv",
+	"image-to-image":                 "cv",
+	"unconditional-image-generation": "cv",
 	"structured-data-classification": "other",
 	"reinforcement-learning":         "other",
 };
@@ -120,6 +124,8 @@ export const PIPELINE_COLOR: {
 	"image-segmentation":             "green",
 	"image-to-text":                  "red",
 	"object-detection":               "orange",
+	"image-to-image":                 "indigo",
+	"unconditional-image-generation": "green",
 	"question-answering":             "blue",
 	"sentence-similarity":            "orange",
 	"structured-data-classification": "indigo",
@@ -167,6 +173,8 @@ export const PIPELINE_TAGS_DISPLAY_ORDER: Array<keyof typeof PipelineType> = [
 	"image-segmentation",
 	"text-to-image",
 	"image-to-text",
+	"image-to-image",
+	"unconditional-image-generation",
 	/// others
 	"structured-data-classification",
 	"reinforcement-learning",

tasks/src/const.ts

@@ -31,6 +31,8 @@ export const TASKS_MODEL_LIBRARIES: Record<keyof typeof PipelineType, Array<keyo
 	"image-segmentation":             ["transformers"],
 	"text-to-image":                  [],
 	"image-to-text":                  [],
+	"image-to-image":                 [],
+	"unconditional-image-generation": [],
 	/// others
 	"structured-data-classification": ["sklearn"],
 	"reinforcement-learning":         ["transformers", "stable-baselines3"],

tasks/src/tasksData.ts

@@ -50,6 +50,8 @@ export const TASKS_DATA: Record<
 	"image-segmentation":             imageSegmentation,
 	"text-to-image":                  undefined,
 	"image-to-text":                  undefined,
+	"image-to-image":                 undefined,
+	"unconditional-image-generation": undefined,
 	/// others
 	"structured-data-classification": undefined,
 	"reinforcement-learning":         undefined,