🚨

packages/tasks/src/index.ts

@@ -2,3 +2,4 @@ export type { TaskData, TaskDemo, TaskDemoEntry, ExampleRepo } from "./Types";
 export { TASKS_DATA } from "./tasksData";
 export { PIPELINE_DATA, type PipelineType, type PipelineData, type Modality, MODALITIES } from "./pipelines";
 export { ModelLibrary } from "./modelLibraries";
+export type { ModelLibraryKey } from "./modelLibraries";

packages/tasks/src/pipelines.ts

@@ -2,6 +2,17 @@ export const MODALITIES = ["cv", "nlp", "audio", "tabular", "multimodal", "rl",
 
 export type Modality = (typeof MODALITIES)[number];
 
+export const MODALITY_LABELS = {
+	multimodal: "Multimodal",
+	nlp: "Natural Language Processing",
+	audio: "Audio",
+	cv: "Computer Vision",
+	rl: "Reinforcement Learning",
+	tabular: "Tabular",
+	other: "Other",
+} satisfies Record<Modality, string>;
+
+
 /**
  * Public interface for a sub task.
  *

packages/widgets/src/lib/components/InferenceWidget/InferenceWidget.svelte

@@ -26,7 +26,7 @@
 	import VisualQuestionAnsweringWidget from "./widgets/VisualQuestionAnsweringWidget/VisualQuestionAnsweringWidget.svelte";
 	import ZeroShotClassificationWidget from "./widgets/ZeroShowClassificationWidget/ZeroShotClassificationWidget.svelte";
 	import ZeroShotImageClassificationWidget from "./widgets/ZeroShotImageClassificationWidget/ZeroShotImageClassificationWidget.svelte";
-	import type { PipelineType } from "$lib/interfaces/Types.js";
+	import type { PipelineType } from "@huggingface/tasks";
 
 	export let apiToken: WidgetProps["apiToken"] = undefined;
 	export let callApiOnMount = false;

packages/widgets/src/lib/components/InferenceWidget/shared/WidgetHeader/WidgetHeader.svelte

@@ -1,10 +1,9 @@
 <script lang="ts">
-	import { TASKS_DATA } from "@huggingface/tasks";
+	import { TASKS_DATA, type PipelineType } from "@huggingface/tasks";
 	import { getPipelineTask } from "../../../../utils/ViewUtils.js";
 	import IconInfo from "$lib/components/Icons/IconInfo.svelte";
 	import IconLightning from "$lib/components/Icons/IconLightning.svelte";
 	import PipelineTag from "../../../PipelineTag/PipelineTag.svelte";
-	import type { PipelineType } from "$lib/interfaces/Types.js";
 
 	export let noTitle = false;
 	export let title: string | null = null;

packages/widgets/src/lib/components/InferenceWidget/widgets/TextGenerationWidget/TextGenerationWidget.svelte

@@ -1,6 +1,5 @@
 <script lang="ts">
 	import type { WidgetProps, ExampleRunOpts, InferenceRunOpts } from "$lib/components/InferenceWidget/shared/types.js";
-	import type { PipelineType } from "$lib/interfaces/Types.js";
 	import type {
 		WidgetExampleTextInput,
 		WidgetExampleOutputText,
@@ -21,6 +20,7 @@
 	} from "$lib/components/InferenceWidget/shared/helpers.js";
 	import { isValidOutputText } from "$lib/components/InferenceWidget/shared/outputValidation.js";
 	import { isTextInput } from "$lib/components/InferenceWidget/shared/inputValidation.js";
+	import type { PipelineType } from "@huggingface/tasks";
 
 	export let apiToken: WidgetProps["apiToken"];
 	export let apiUrl: WidgetProps["apiUrl"];

packages/widgets/src/lib/components/PipelineIcon/PipelineIcon.svelte

@@ -1,6 +1,5 @@
 <script lang="ts">
 	import type { SvelteComponent } from "svelte";
-	import type { PipelineType } from "../../interfaces/Types.js";
 
 	import IconAudioClassification from "../Icons/IconAudioClassification.svelte";
 	import IconAudioToAudio from "../Icons/IconAudioToAudio.svelte";
@@ -36,6 +35,7 @@
 	import IconUnconditionalImageGeneration from "../Icons/IconUnconditionalImageGeneration.svelte";
 	import IconDocumentQuestionAnswering from "../Icons/IconDocumentQuestionAnswering.svelte";
 	import IconGraphML from "../Icons/IconGraphML.svelte";
+	import type { PipelineType } from "@huggingface/tasks";
 
 	export let classNames = "";
 	export let pipeline = "";

packages/widgets/src/lib/inferenceSnippets/inputs.ts

@@ -1,4 +1,5 @@
-import type { PipelineType, ModelData } from "../interfaces/Types.js";
+import type { PipelineType } from "@huggingface/tasks";
+import type { ModelData } from "../interfaces/Types.js";
 
 const inputsZeroShotClassification = () =>
 	`"Hi, I recently bought a device from your company but it is not working as advertised and I would like to get reimbursed!"`;

packages/widgets/src/lib/inferenceSnippets/serveCurl.ts

@@ -1,4 +1,5 @@
-import type { PipelineType, ModelData } from "../interfaces/Types.js";
+import type { PipelineType } from "@huggingface/tasks";
+import type { ModelData } from "../interfaces/Types.js";
 import { getModelInputSnippet } from "./inputs.js";
 
 export const snippetBasic = (model: ModelData, accessToken: string): string =>

packages/widgets/src/lib/inferenceSnippets/serveJs.ts

@@ -1,4 +1,5 @@
-import type { PipelineType, ModelData } from "../interfaces/Types.js";
+import type { PipelineType } from "@huggingface/tasks";
+import type { ModelData } from "../interfaces/Types.js";
 import { getModelInputSnippet } from "./inputs.js";
 
 export const snippetBasic = (model: ModelData, accessToken: string): string =>

packages/widgets/src/lib/inferenceSnippets/servePython.ts

@@ -1,5 +1,5 @@
+import type { PipelineType } from "@huggingface/tasks";
 import type { ModelData } from "../interfaces/Types.js";
-import type { PipelineType } from "../interfaces/Types.js";
 import { getModelInputSnippet } from "./inputs.js";
 
 export const snippetZeroShotClassification = (model: ModelData): string =>

packages/widgets/src/lib/interfaces/DefaultWidget.ts

@@ -1,5 +1,5 @@
+import type { PipelineType } from "@huggingface/tasks";
 import type { WidgetExample } from "../components/InferenceWidget/shared/WidgetExample.js";
-import type { PipelineType } from "./Types.js";
 
 type LanguageCode = string;
 

packages/widgets/src/lib/interfaces/Libraries.ts

@@ -1,57 +1,7 @@
+import type { ModelLibraryKey } from "@huggingface/tasks";
 /* eslint-disable @typescript-eslint/naming-convention */
 import type { ModelData } from "./Types.js";
 
-/**
- * Add your new library here.
- *
- * This is for modeling (= architectures) libraries, not for file formats (like ONNX, etc).
- * File formats live in an enum inside the internal codebase.
- */
-export enum ModelLibrary {
-	"adapter-transformers" = "Adapter Transformers",
-	"allennlp" = "allenNLP",
-	"asteroid" = "Asteroid",
-	"bertopic" = "BERTopic",
-	"diffusers" = "Diffusers",
-	"doctr" = "docTR",
-	"espnet" = "ESPnet",
-	"fairseq" = "Fairseq",
-	"flair" = "Flair",
-	"keras" = "Keras",
-	"k2" = "K2",
-	"nemo" = "NeMo",
-	"open_clip" = "OpenCLIP",
-	"paddlenlp" = "PaddleNLP",
-	"peft" = "PEFT",
-	"pyannote-audio" = "pyannote.audio",
-	"sample-factory" = "Sample Factory",
-	"sentence-transformers" = "Sentence Transformers",
-	"sklearn" = "Scikit-learn",
-	"spacy" = "spaCy",
-	"span-marker" = "SpanMarker",
-	"speechbrain" = "speechbrain",
-	"tensorflowtts" = "TensorFlowTTS",
-	"timm" = "Timm",
-	"fastai" = "fastai",
-	"transformers" = "Transformers",
-	"transformers.js" = "Transformers.js",
-	"stanza" = "Stanza",
-	"fasttext" = "fastText",
-	"stable-baselines3" = "Stable-Baselines3",
-	"ml-agents" = "ML-Agents",
-	"pythae" = "Pythae",
-	"mindspore" = "MindSpore",
-}
-
-export type ModelLibraryKey = keyof typeof ModelLibrary;
-export const ALL_MODEL_LIBRARY_KEYS = Object.keys(ModelLibrary) as ModelLibraryKey[];
-
-const EXCLUDE_THOSE_LIBRARIES_FROM_DISPLAY: ModelLibraryKey[] = ["doctr", "k2", "mindspore", "tensorflowtts"];
-
-export const ALL_DISPLAY_MODEL_LIBRARY_KEYS = ALL_MODEL_LIBRARY_KEYS.filter(
-	(k) => !EXCLUDE_THOSE_LIBRARIES_FROM_DISPLAY.includes(k)
-);
-
 /**
  * Elements configurable by a model library.
  */

packages/widgets/src/lib/interfaces/LibrariesToTasks.ts

@@ -1,5 +1,4 @@
-import type { ModelLibraryKey } from "./Libraries.js";
-import type { PipelineType } from "./Types.js";
+import type { ModelLibraryKey, PipelineType } from "@huggingface/tasks";
 
 /**
  * Mapping from library name (excluding Transformers) to its supported tasks.