Sync widgets demo

packages/tasks/src/pipelines.ts

@@ -435,6 +435,20 @@ export const PIPELINE_DATA = {
 	},
 	"image-to-image": {
 		name: "Image-to-Image",
+		subtasks: [
+			{
+				type: "image-inpainting",
+				name: "Image Inpainting",
+			},
+			{
+				type: "image-colorization",
+				name: "Image Colorization",
+			},
+			{
+				type: "super-resolution",
+				name: "Super Resolution",
+			},
+		],
 		modality: "cv",
 		color: "indigo",
 	},

packages/tasks/src/tasks/index.ts

@@ -40,18 +40,18 @@ import type { ModelLibraryKey } from "../model-libraries";
  * Model libraries compatible with each ML task
  */
 export const TASKS_MODEL_LIBRARIES: Record<PipelineType, ModelLibraryKey[]> = {
-	"audio-classification": ["speechbrain", "transformers"],
+	"audio-classification": ["speechbrain", "transformers", "transformers.js"],
 	"audio-to-audio": ["asteroid", "speechbrain"],
 	"automatic-speech-recognition": ["espnet", "nemo", "speechbrain", "transformers", "transformers.js"],
 	conversational: ["transformers"],
-	"depth-estimation": ["transformers"],
-	"document-question-answering": ["transformers"],
+	"depth-estimation": ["transformers", "transformers.js"],
+	"document-question-answering": ["transformers", "transformers.js"],
 	"feature-extraction": ["sentence-transformers", "transformers", "transformers.js"],
 	"fill-mask": ["transformers", "transformers.js"],
 	"graph-ml": ["transformers"],
 	"image-classification": ["keras", "timm", "transformers", "transformers.js"],
 	"image-segmentation": ["transformers", "transformers.js"],
-	"image-to-image": ["diffusers"],
+	"image-to-image": ["diffusers", "transformers.js"],
 	"image-to-text": ["transformers.js"],
 	"image-to-video": ["diffusers"],
 	"video-classification": ["transformers"],
@@ -73,8 +73,8 @@ export const TASKS_MODEL_LIBRARIES: Record<PipelineType, ModelLibraryKey[]> = {
 	"text-generation": ["transformers", "transformers.js"],
 	"text-retrieval": [],
 	"text-to-image": ["diffusers"],
-	"text-to-speech": ["espnet", "tensorflowtts", "transformers"],
-	"text-to-audio": ["transformers"],
+	"text-to-speech": ["espnet", "tensorflowtts", "transformers", "transformers.js"],
+	"text-to-audio": ["transformers", "transformers.js"],
 	"text-to-video": ["diffusers"],
 	"text2text-generation": ["transformers", "transformers.js"],
 	"time-series-forecasting": [],
@@ -89,11 +89,11 @@ export const TASKS_MODEL_LIBRARIES: Record<PipelineType, ModelLibraryKey[]> = {
 	],
 	translation: ["transformers", "transformers.js"],
 	"unconditional-image-generation": ["diffusers"],
-	"visual-question-answering": ["transformers"],
+	"visual-question-answering": ["transformers", "transformers.js"],
 	"voice-activity-detection": [],
 	"zero-shot-classification": ["transformers", "transformers.js"],
 	"zero-shot-image-classification": ["transformers", "transformers.js"],
-	"zero-shot-object-detection": ["transformers"],
+	"zero-shot-object-detection": ["transformers", "transformers.js"],
 };
 
 /**