diff --git a/.gitignore b/.gitignore
index c938cddaf..01b907de2 100644
--- a/.gitignore
+++ b/.gitignore
@@ -14,7 +14,6 @@ dist/
downloads/
eggs/
.eggs/
-lib/
lib64/
parts/
sdist/
diff --git a/js/src/lib/components/Icons/IconImageToImage.svelte b/js/src/lib/components/Icons/IconImageToImage.svelte
new file mode 100644
index 000000000..1ea332d30
--- /dev/null
+++ b/js/src/lib/components/Icons/IconImageToImage.svelte
@@ -0,0 +1,25 @@
+
+
+
diff --git a/js/src/lib/components/Icons/IconUnconditionalImageGeneration.svelte b/js/src/lib/components/Icons/IconUnconditionalImageGeneration.svelte
new file mode 100644
index 000000000..87aec18e6
--- /dev/null
+++ b/js/src/lib/components/Icons/IconUnconditionalImageGeneration.svelte
@@ -0,0 +1,25 @@
+
+
+
diff --git a/js/src/lib/components/ModelPipelineIcon/ModelPipelineIcon.svelte b/js/src/lib/components/ModelPipelineIcon/ModelPipelineIcon.svelte
index 2a14b72a9..1595168b7 100644
--- a/js/src/lib/components/ModelPipelineIcon/ModelPipelineIcon.svelte
+++ b/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,
};
diff --git a/js/src/lib/interfaces/Types.ts b/js/src/lib/interfaces/Types.ts
index 0ab71d7b7..c64086f86 100644
--- a/js/src/lib/interfaces/Types.ts
+++ b/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 = [
"image-segmentation",
"text-to-image",
"image-to-text",
+ "image-to-image",
+ "unconditional-image-generation",
/// others
"structured-data-classification",
"reinforcement-learning",
diff --git a/tasks/src/const.ts b/tasks/src/const.ts
index d00691d20..c4a052e05 100644
--- a/tasks/src/const.ts
+++ b/tasks/src/const.ts
@@ -31,6 +31,8 @@ export const TASKS_MODEL_LIBRARIES: Record