@huggingface/tasks 0.9.0 → 0.10.0

package/dist/src/pipelines.d.ts

@@ -0,0 +1,404 @@
+export declare const MODALITIES: readonly ["cv", "nlp", "audio", "tabular", "multimodal", "rl", "other"];
+export type Modality = (typeof MODALITIES)[number];
+export declare const MODALITY_LABELS: {
+    multimodal: string;
+    nlp: string;
+    audio: string;
+    cv: string;
+    rl: string;
+    tabular: string;
+    other: string;
+};
+/**
+ * Public interface for a sub task.
+ *
+ * This can be used in a model card's `model-index` metadata.
+ * and is more granular classification that can grow significantly
+ * over time as new tasks are added.
+ */
+export interface SubTask {
+    /**
+     * type of the task (e.g. audio-source-separation)
+     */
+    type: string;
+    /**
+     * displayed name of the task (e.g. Audio Source Separation)
+     */
+    name: string;
+}
+/**
+ * Public interface for a PipelineData.
+ *
+ * This information corresponds to a pipeline type (aka task)
+ * in the Hub.
+ */
+export interface PipelineData {
+    /**
+     * displayed name of the task (e.g. Text Classification)
+     */
+    name: string;
+    subtasks?: SubTask[];
+    modality: Modality;
+    /**
+     * color for the tag icon.
+     */
+    color: "blue" | "green" | "indigo" | "orange" | "red" | "yellow";
+    /**
+     * whether to hide in /models filters
+     */
+    hideInModels?: boolean;
+    /**
+     * whether to hide in /datasets filters
+     */
+    hideInDatasets?: boolean;
+}
+export declare const PIPELINE_DATA: {
+    "text-classification": {
+        name: string;
+        subtasks: {
+            type: string;
+            name: string;
+        }[];
+        modality: "nlp";
+        color: "orange";
+    };
+    "token-classification": {
+        name: string;
+        subtasks: {
+            type: string;
+            name: string;
+        }[];
+        modality: "nlp";
+        color: "blue";
+    };
+    "table-question-answering": {
+        name: string;
+        modality: "nlp";
+        color: "green";
+    };
+    "question-answering": {
+        name: string;
+        subtasks: {
+            type: string;
+            name: string;
+        }[];
+        modality: "nlp";
+        color: "blue";
+    };
+    "zero-shot-classification": {
+        name: string;
+        modality: "nlp";
+        color: "yellow";
+    };
+    translation: {
+        name: string;
+        modality: "nlp";
+        color: "green";
+    };
+    summarization: {
+        name: string;
+        subtasks: {
+            type: string;
+            name: string;
+        }[];
+        modality: "nlp";
+        color: "indigo";
+    };
+    "feature-extraction": {
+        name: string;
+        modality: "nlp";
+        color: "red";
+    };
+    "text-generation": {
+        name: string;
+        subtasks: {
+            type: string;
+            name: string;
+        }[];
+        modality: "nlp";
+        color: "indigo";
+    };
+    "text2text-generation": {
+        name: string;
+        subtasks: {
+            type: string;
+            name: string;
+        }[];
+        modality: "nlp";
+        color: "indigo";
+    };
+    "fill-mask": {
+        name: string;
+        subtasks: {
+            type: string;
+            name: string;
+        }[];
+        modality: "nlp";
+        color: "red";
+    };
+    "sentence-similarity": {
+        name: string;
+        modality: "nlp";
+        color: "yellow";
+    };
+    "text-to-speech": {
+        name: string;
+        modality: "audio";
+        color: "yellow";
+    };
+    "text-to-audio": {
+        name: string;
+        modality: "audio";
+        color: "yellow";
+    };
+    "automatic-speech-recognition": {
+        name: string;
+        modality: "audio";
+        color: "yellow";
+    };
+    "audio-to-audio": {
+        name: string;
+        modality: "audio";
+        color: "blue";
+    };
+    "audio-classification": {
+        name: string;
+        subtasks: {
+            type: string;
+            name: string;
+        }[];
+        modality: "audio";
+        color: "green";
+    };
+    "voice-activity-detection": {
+        name: string;
+        modality: "audio";
+        color: "red";
+    };
+    "depth-estimation": {
+        name: string;
+        modality: "cv";
+        color: "yellow";
+    };
+    "image-classification": {
+        name: string;
+        subtasks: {
+            type: string;
+            name: string;
+        }[];
+        modality: "cv";
+        color: "blue";
+    };
+    "object-detection": {
+        name: string;
+        subtasks: {
+            type: string;
+            name: string;
+        }[];
+        modality: "cv";
+        color: "yellow";
+    };
+    "image-segmentation": {
+        name: string;
+        subtasks: {
+            type: string;
+            name: string;
+        }[];
+        modality: "cv";
+        color: "green";
+    };
+    "text-to-image": {
+        name: string;
+        modality: "cv";
+        color: "yellow";
+    };
+    "image-to-text": {
+        name: string;
+        subtasks: {
+            type: string;
+            name: string;
+        }[];
+        modality: "cv";
+        color: "red";
+    };
+    "image-to-image": {
+        name: string;
+        subtasks: {
+            type: string;
+            name: string;
+        }[];
+        modality: "cv";
+        color: "indigo";
+    };
+    "image-to-video": {
+        name: string;
+        modality: "cv";
+        color: "indigo";
+    };
+    "unconditional-image-generation": {
+        name: string;
+        modality: "cv";
+        color: "green";
+    };
+    "video-classification": {
+        name: string;
+        modality: "cv";
+        color: "blue";
+    };
+    "reinforcement-learning": {
+        name: string;
+        modality: "rl";
+        color: "red";
+    };
+    robotics: {
+        name: string;
+        modality: "rl";
+        subtasks: {
+            type: string;
+            name: string;
+        }[];
+        color: "blue";
+    };
+    "tabular-classification": {
+        name: string;
+        modality: "tabular";
+        subtasks: {
+            type: string;
+            name: string;
+        }[];
+        color: "blue";
+    };
+    "tabular-regression": {
+        name: string;
+        modality: "tabular";
+        subtasks: {
+            type: string;
+            name: string;
+        }[];
+        color: "blue";
+    };
+    "tabular-to-text": {
+        name: string;
+        modality: "tabular";
+        subtasks: {
+            type: string;
+            name: string;
+        }[];
+        color: "blue";
+        hideInModels: true;
+    };
+    "table-to-text": {
+        name: string;
+        modality: "nlp";
+        color: "blue";
+        hideInModels: true;
+    };
+    "multiple-choice": {
+        name: string;
+        subtasks: {
+            type: string;
+            name: string;
+        }[];
+        modality: "nlp";
+        color: "blue";
+        hideInModels: true;
+    };
+    "text-retrieval": {
+        name: string;
+        subtasks: {
+            type: string;
+            name: string;
+        }[];
+        modality: "nlp";
+        color: "indigo";
+        hideInModels: true;
+    };
+    "time-series-forecasting": {
+        name: string;
+        modality: "tabular";
+        subtasks: {
+            type: string;
+            name: string;
+        }[];
+        color: "blue";
+        hideInModels: true;
+    };
+    "text-to-video": {
+        name: string;
+        modality: "cv";
+        color: "green";
+    };
+    "image-text-to-text": {
+        name: string;
+        modality: "multimodal";
+        color: "red";
+        hideInDatasets: true;
+    };
+    "visual-question-answering": {
+        name: string;
+        subtasks: {
+            type: string;
+            name: string;
+        }[];
+        modality: "multimodal";
+        color: "red";
+    };
+    "document-question-answering": {
+        name: string;
+        subtasks: {
+            type: string;
+            name: string;
+        }[];
+        modality: "multimodal";
+        color: "blue";
+        hideInDatasets: true;
+    };
+    "zero-shot-image-classification": {
+        name: string;
+        modality: "cv";
+        color: "yellow";
+    };
+    "graph-ml": {
+        name: string;
+        modality: "other";
+        color: "green";
+    };
+    "mask-generation": {
+        name: string;
+        modality: "cv";
+        color: "indigo";
+    };
+    "zero-shot-object-detection": {
+        name: string;
+        modality: "cv";
+        color: "yellow";
+    };
+    "text-to-3d": {
+        name: string;
+        modality: "cv";
+        color: "yellow";
+    };
+    "image-to-3d": {
+        name: string;
+        modality: "cv";
+        color: "green";
+    };
+    "image-feature-extraction": {
+        name: string;
+        modality: "cv";
+        color: "indigo";
+    };
+    other: {
+        name: string;
+        modality: "other";
+        color: "blue";
+        hideInModels: true;
+        hideInDatasets: true;
+    };
+};
+export type PipelineType = keyof typeof PIPELINE_DATA;
+export type WidgetType = PipelineType | "conversational";
+export declare const PIPELINE_TYPES: ("other" | "text-classification" | "token-classification" | "table-question-answering" | "question-answering" | "zero-shot-classification" | "translation" | "summarization" | "feature-extraction" | "text-generation" | "text2text-generation" | "fill-mask" | "sentence-similarity" | "text-to-speech" | "text-to-audio" | "automatic-speech-recognition" | "audio-to-audio" | "audio-classification" | "voice-activity-detection" | "depth-estimation" | "image-classification" | "object-detection" | "image-segmentation" | "text-to-image" | "image-to-text" | "image-to-image" | "image-to-video" | "unconditional-image-generation" | "video-classification" | "reinforcement-learning" | "robotics" | "tabular-classification" | "tabular-regression" | "tabular-to-text" | "table-to-text" | "multiple-choice" | "text-retrieval" | "time-series-forecasting" | "text-to-video" | "image-text-to-text" | "visual-question-answering" | "document-question-answering" | "zero-shot-image-classification" | "graph-ml" | "mask-generation" | "zero-shot-object-detection" | "text-to-3d" | "image-to-3d" | "image-feature-extraction")[];
+export declare const SUBTASK_TYPES: string[];
+export declare const PIPELINE_TYPES_SET: Set<"other" | "text-classification" | "token-classification" | "table-question-answering" | "question-answering" | "zero-shot-classification" | "translation" | "summarization" | "feature-extraction" | "text-generation" | "text2text-generation" | "fill-mask" | "sentence-similarity" | "text-to-speech" | "text-to-audio" | "automatic-speech-recognition" | "audio-to-audio" | "audio-classification" | "voice-activity-detection" | "depth-estimation" | "image-classification" | "object-detection" | "image-segmentation" | "text-to-image" | "image-to-text" | "image-to-image" | "image-to-video" | "unconditional-image-generation" | "video-classification" | "reinforcement-learning" | "robotics" | "tabular-classification" | "tabular-regression" | "tabular-to-text" | "table-to-text" | "multiple-choice" | "text-retrieval" | "time-series-forecasting" | "text-to-video" | "image-text-to-text" | "visual-question-answering" | "document-question-answering" | "zero-shot-image-classification" | "graph-ml" | "mask-generation" | "zero-shot-object-detection" | "text-to-3d" | "image-to-3d" | "image-feature-extraction">;
+//# sourceMappingURL=pipelines.d.ts.map

package/dist/src/pipelines.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"pipelines.d.ts","sourceRoot":"","sources":["../../src/pipelines.ts"],"names":[],"mappings":"AAAA,eAAO,MAAM,UAAU,yEAA0E,CAAC;AAElG,MAAM,MAAM,QAAQ,GAAG,CAAC,OAAO,UAAU,CAAC,CAAC,MAAM,CAAC,CAAC;AAEnD,eAAO,MAAM,eAAe;;;;;;;;CAQQ,CAAC;AAErC;;;;;;GAMG;AACH,MAAM,WAAW,OAAO;IACvB;;OAEG;IACH,IAAI,EAAE,MAAM,CAAC;IACb;;OAEG;IACH,IAAI,EAAE,MAAM,CAAC;CACb;AAED;;;;;GAKG;AACH,MAAM,WAAW,YAAY;IAC5B;;OAEG;IACH,IAAI,EAAE,MAAM,CAAC;IACb,QAAQ,CAAC,EAAE,OAAO,EAAE,CAAC;IACrB,QAAQ,EAAE,QAAQ,CAAC;IACnB;;OAEG;IACH,KAAK,EAAE,MAAM,GAAG,OAAO,GAAG,QAAQ,GAAG,QAAQ,GAAG,KAAK,GAAG,QAAQ,CAAC;IACjE;;OAEG;IACH,YAAY,CAAC,EAAE,OAAO,CAAC;IACvB;;OAEG;IACH,cAAc,CAAC,EAAE,OAAO,CAAC;CACzB;AAcD,eAAO,MAAM,aAAa;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CA+kBc,CAAC;AAEzC,MAAM,MAAM,YAAY,GAAG,MAAM,OAAO,aAAa,CAAC;AAEtD,MAAM,MAAM,UAAU,GAAG,YAAY,GAAG,gBAAgB,CAAC;AAEzD,eAAO,MAAM,cAAc,mlCAA+C,CAAC;AAE3E,eAAO,MAAM,aAAa,UAEN,CAAC;AAErB,eAAO,MAAM,kBAAkB,olCAA0B,CAAC"}

package/dist/src/snippets/curl.d.ts

@@ -0,0 +1,9 @@
+import type { PipelineType } from "../pipelines.js";
+import type { ModelDataMinimal } from "./types.js";
+export declare const snippetBasic: (model: ModelDataMinimal, accessToken: string) => string;
+export declare const snippetZeroShotClassification: (model: ModelDataMinimal, accessToken: string) => string;
+export declare const snippetFile: (model: ModelDataMinimal, accessToken: string) => string;
+export declare const curlSnippets: Partial<Record<PipelineType, (model: ModelDataMinimal, accessToken: string) => string>>;
+export declare function getCurlInferenceSnippet(model: ModelDataMinimal, accessToken: string): string;
+export declare function hasCurlInferenceSnippet(model: Pick<ModelDataMinimal, "pipeline_tag">): boolean;
+//# sourceMappingURL=curl.d.ts.map

package/dist/src/snippets/curl.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"curl.d.ts","sourceRoot":"","sources":["../../../src/snippets/curl.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,iBAAiB,CAAC;AAEpD,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,YAAY,CAAC;AAEnD,eAAO,MAAM,YAAY,UAAW,gBAAgB,eAAe,MAAM,KAAG,MAM3E,CAAC;AAEF,eAAO,MAAM,6BAA6B,UAAW,gBAAgB,eAAe,MAAM,KAAG,MAM5F,CAAC;AAEF,eAAO,MAAM,WAAW,UAAW,gBAAgB,eAAe,MAAM,KAAG,MAK1E,CAAC;AAEF,eAAO,MAAM,YAAY,EAAE,OAAO,CAAC,MAAM,CAAC,YAAY,EAAE,CAAC,KAAK,EAAE,gBAAgB,EAAE,WAAW,EAAE,MAAM,KAAK,MAAM,CAAC,CAwBhH,CAAC;AAEF,wBAAgB,uBAAuB,CAAC,KAAK,EAAE,gBAAgB,EAAE,WAAW,EAAE,MAAM,GAAG,MAAM,CAI5F;AAED,wBAAgB,uBAAuB,CAAC,KAAK,EAAE,IAAI,CAAC,gBAAgB,EAAE,cAAc,CAAC,GAAG,OAAO,CAE9F"}

package/dist/src/snippets/index.d.ts

@@ -0,0 +1,6 @@
+import * as inputs from "./inputs";
+import * as curl from "./curl";
+import * as python from "./python";
+import * as js from "./js";
+export { inputs, curl, python, js };
+//# sourceMappingURL=index.d.ts.map

package/dist/src/snippets/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/snippets/index.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,MAAM,MAAM,UAAU,CAAC;AACnC,OAAO,KAAK,IAAI,MAAM,QAAQ,CAAC;AAC/B,OAAO,KAAK,MAAM,MAAM,UAAU,CAAC;AACnC,OAAO,KAAK,EAAE,MAAM,MAAM,CAAC;AAE3B,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC"}

package/dist/src/snippets/inputs.d.ts

@@ -0,0 +1,3 @@
+import type { ModelDataMinimal } from "./types";
+export declare function getModelInputSnippet(model: ModelDataMinimal, noWrap?: boolean, noQuotes?: boolean): string;
+//# sourceMappingURL=inputs.d.ts.map

package/dist/src/snippets/inputs.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"inputs.d.ts","sourceRoot":"","sources":["../../../src/snippets/inputs.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,SAAS,CAAC;AAqHhD,wBAAgB,oBAAoB,CAAC,KAAK,EAAE,gBAAgB,EAAE,MAAM,UAAQ,EAAE,QAAQ,UAAQ,GAAG,MAAM,CAiBtG"}

package/dist/src/snippets/js.d.ts

@@ -0,0 +1,11 @@
+import type { PipelineType } from "../pipelines.js";
+import type { ModelDataMinimal } from "./types.js";
+export declare const snippetBasic: (model: ModelDataMinimal, accessToken: string) => string;
+export declare const snippetZeroShotClassification: (model: ModelDataMinimal, accessToken: string) => string;
+export declare const snippetTextToImage: (model: ModelDataMinimal, accessToken: string) => string;
+export declare const snippetTextToAudio: (model: ModelDataMinimal, accessToken: string) => string;
+export declare const snippetFile: (model: ModelDataMinimal, accessToken: string) => string;
+export declare const jsSnippets: Partial<Record<PipelineType, (model: ModelDataMinimal, accessToken: string) => string>>;
+export declare function getJsInferenceSnippet(model: ModelDataMinimal, accessToken: string): string;
+export declare function hasJsInferenceSnippet(model: ModelDataMinimal): boolean;
+//# sourceMappingURL=js.d.ts.map

package/dist/src/snippets/js.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"js.d.ts","sourceRoot":"","sources":["../../../src/snippets/js.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,iBAAiB,CAAC;AAEpD,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,YAAY,CAAC;AAEnD,eAAO,MAAM,YAAY,UAAW,gBAAgB,eAAe,MAAM,KAAG,MAgBxE,CAAC;AAEL,eAAO,MAAM,6BAA6B,UAAW,gBAAgB,eAAe,MAAM,KAAG,MAkBzF,CAAC;AAEL,eAAO,MAAM,kBAAkB,UAAW,gBAAgB,eAAe,MAAM,KAAG,MAe9E,CAAC;AAEL,eAAO,MAAM,kBAAkB,UAAW,gBAAgB,eAAe,MAAM,KAAG,MAkCjF,CAAC;AAEF,eAAO,MAAM,WAAW,UAAW,gBAAgB,eAAe,MAAM,KAAG,MAiBvE,CAAC;AAEL,eAAO,MAAM,UAAU,EAAE,OAAO,CAAC,MAAM,CAAC,YAAY,EAAE,CAAC,KAAK,EAAE,gBAAgB,EAAE,WAAW,EAAE,MAAM,KAAK,MAAM,CAAC,CAwB9G,CAAC;AAEF,wBAAgB,qBAAqB,CAAC,KAAK,EAAE,gBAAgB,EAAE,WAAW,EAAE,MAAM,GAAG,MAAM,CAI1F;AAED,wBAAgB,qBAAqB,CAAC,KAAK,EAAE,gBAAgB,GAAG,OAAO,CAEtE"}

package/dist/src/snippets/python.d.ts

@@ -0,0 +1,14 @@
+import type { PipelineType } from "../pipelines.js";
+import type { ModelDataMinimal } from "./types.js";
+export declare const snippetZeroShotClassification: (model: ModelDataMinimal) => string;
+export declare const snippetZeroShotImageClassification: (model: ModelDataMinimal) => string;
+export declare const snippetBasic: (model: ModelDataMinimal) => string;
+export declare const snippetFile: (model: ModelDataMinimal) => string;
+export declare const snippetTextToImage: (model: ModelDataMinimal) => string;
+export declare const snippetTabular: (model: ModelDataMinimal) => string;
+export declare const snippetTextToAudio: (model: ModelDataMinimal) => string;
+export declare const snippetDocumentQuestionAnswering: (model: ModelDataMinimal) => string;
+export declare const pythonSnippets: Partial<Record<PipelineType, (model: ModelDataMinimal) => string>>;
+export declare function getPythonInferenceSnippet(model: ModelDataMinimal, accessToken: string): string;
+export declare function hasPythonInferenceSnippet(model: ModelDataMinimal): boolean;
+//# sourceMappingURL=python.d.ts.map

package/dist/src/snippets/python.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"python.d.ts","sourceRoot":"","sources":["../../../src/snippets/python.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,iBAAiB,CAAC;AAEpD,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,YAAY,CAAC;AAEnD,eAAO,MAAM,6BAA6B,UAAW,gBAAgB,KAAG,MAQrE,CAAC;AAEJ,eAAO,MAAM,kCAAkC,UAAW,gBAAgB,KAAG,MAc1E,CAAC;AAEJ,eAAO,MAAM,YAAY,UAAW,gBAAgB,KAAG,MAOpD,CAAC;AAEJ,eAAO,MAAM,WAAW,UAAW,gBAAgB,KAAG,MAOP,CAAC;AAEhD,eAAO,MAAM,kBAAkB,UAAW,gBAAgB,KAAG,MAUjB,CAAC;AAE7C,eAAO,MAAM,cAAc,UAAW,gBAAgB,KAAG,MAMtD,CAAC;AAEJ,eAAO,MAAM,kBAAkB,UAAW,gBAAgB,KAAG,MA2B5D,CAAC;AAEF,eAAO,MAAM,gCAAgC,UAAW,gBAAgB,KAAG,MAUxE,CAAC;AAEJ,eAAO,MAAM,cAAc,EAAE,OAAO,CAAC,MAAM,CAAC,YAAY,EAAE,CAAC,KAAK,EAAE,gBAAgB,KAAK,MAAM,CAAC,CA4B7F,CAAC;AAEF,wBAAgB,yBAAyB,CAAC,KAAK,EAAE,gBAAgB,EAAE,WAAW,EAAE,MAAM,GAAG,MAAM,CAU9F;AAED,wBAAgB,yBAAyB,CAAC,KAAK,EAAE,gBAAgB,GAAG,OAAO,CAE1E"}

package/dist/src/snippets/types.d.ts

@@ -0,0 +1,8 @@
+import type { ModelData } from "../model-data";
+/**
+ * Minimal model data required for snippets.
+ *
+ * Add more fields as needed.
+ */
+export type ModelDataMinimal = Pick<ModelData, "id" | "pipeline_tag" | "mask_token" | "library_name">;
+//# sourceMappingURL=types.d.ts.map

package/dist/src/snippets/types.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../../src/snippets/types.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,eAAe,CAAC;AAE/C;;;;GAIG;AACH,MAAM,MAAM,gBAAgB,GAAG,IAAI,CAAC,SAAS,EAAE,IAAI,GAAG,cAAc,GAAG,YAAY,GAAG,cAAc,CAAC,CAAC"}

package/dist/src/tasks/audio-classification/data.d.ts

@@ -0,0 +1,4 @@
+import type { TaskDataCustom } from "..";
+declare const taskData: TaskDataCustom;
+export default taskData;
+//# sourceMappingURL=data.d.ts.map

package/dist/src/tasks/audio-classification/data.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"data.d.ts","sourceRoot":"","sources":["../../../../src/tasks/audio-classification/data.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,IAAI,CAAC;AAEzC,QAAA,MAAM,QAAQ,EAAE,cAwEf,CAAC;AAEF,eAAe,QAAQ,CAAC"}

package/dist/src/tasks/audio-classification/inference.d.ts

@@ -0,0 +1,52 @@
+/**
+ * Inference code generated from the JSON schema spec in ./spec
+ *
+ * Using src/scripts/inference-codegen
+ */
+/**
+ * Inputs for Audio Classification inference
+ */
+export interface AudioClassificationInput {
+    /**
+     * The input audio data
+     */
+    inputs: unknown;
+    /**
+     * Additional inference parameters
+     */
+    parameters?: AudioClassificationParameters;
+    [property: string]: unknown;
+}
+/**
+ * Additional inference parameters
+ *
+ * Additional inference parameters for Audio Classification
+ */
+export interface AudioClassificationParameters {
+    function_to_apply?: ClassificationOutputTransform;
+    /**
+     * When specified, limits the output to the top K most probable classes.
+     */
+    top_k?: number;
+    [property: string]: unknown;
+}
+/**
+ * The function to apply to the model outputs in order to retrieve the scores.
+ */
+export type ClassificationOutputTransform = "sigmoid" | "softmax" | "none";
+export type AudioClassificationOutput = AudioClassificationOutputElement[];
+/**
+ * Outputs for Audio Classification inference
+ */
+export interface AudioClassificationOutputElement {
+    /**
+     * The predicted class label.
+     */
+    label: string;
+    /**
+     * The corresponding probability.
+     */
+    score: number;
+    [property: string]: unknown;
+}
+//# sourceMappingURL=inference.d.ts.map

package/dist/src/tasks/audio-classification/inference.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"inference.d.ts","sourceRoot":"","sources":["../../../../src/tasks/audio-classification/inference.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AACH;;GAEG;AACH,MAAM,WAAW,wBAAwB;IACxC;;OAEG;IACH,MAAM,EAAE,OAAO,CAAC;IAChB;;OAEG;IACH,UAAU,CAAC,EAAE,6BAA6B,CAAC;IAC3C,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B;AACD;;;;GAIG;AACH,MAAM,WAAW,6BAA6B;IAC7C,iBAAiB,CAAC,EAAE,6BAA6B,CAAC;IAClD;;OAEG;IACH,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B;AACD;;GAEG;AACH,MAAM,MAAM,6BAA6B,GAAG,SAAS,GAAG,SAAS,GAAG,MAAM,CAAC;AAC3E,MAAM,MAAM,yBAAyB,GAAG,gCAAgC,EAAE,CAAC;AAC3E;;GAEG;AACH,MAAM,WAAW,gCAAgC;IAChD;;OAEG;IACH,KAAK,EAAE,MAAM,CAAC;IACd;;OAEG;IACH,KAAK,EAAE,MAAM,CAAC;IACd,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B"}

package/dist/src/tasks/audio-to-audio/data.d.ts

@@ -0,0 +1,4 @@
+import type { TaskDataCustom } from "..";
+declare const taskData: TaskDataCustom;
+export default taskData;
+//# sourceMappingURL=data.d.ts.map

package/dist/src/tasks/audio-to-audio/data.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"data.d.ts","sourceRoot":"","sources":["../../../../src/tasks/audio-to-audio/data.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,IAAI,CAAC;AAEzC,QAAA,MAAM,QAAQ,EAAE,cA6Df,CAAC;AAEF,eAAe,QAAQ,CAAC"}

package/dist/src/tasks/automatic-speech-recognition/data.d.ts

@@ -0,0 +1,4 @@
+import type { TaskDataCustom } from "..";
+declare const taskData: TaskDataCustom;
+export default taskData;
+//# sourceMappingURL=data.d.ts.map

package/dist/src/tasks/automatic-speech-recognition/data.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"data.d.ts","sourceRoot":"","sources":["../../../../src/tasks/automatic-speech-recognition/data.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,IAAI,CAAC;AAEzC,QAAA,MAAM,QAAQ,EAAE,cAyEf,CAAC;AAEF,eAAe,QAAQ,CAAC"}