@huggingface/tasks 0.13.0-test2 → 0.13.0-test4

package/dist/tasks/depth-estimation/inference.d.ts

@@ -0,0 +1,36 @@
+/**
+ * Inference code generated from the JSON schema spec in ./spec
+ *
+ * Using src/scripts/inference-codegen
+ */
+/**
+ * Inputs for Depth Estimation inference
+ */
+export interface DepthEstimationInput {
+    /**
+     * The input image data
+     */
+    inputs: unknown;
+    /**
+     * Additional inference parameters
+     */
+    parameters?: {
+        [key: string]: unknown;
+    };
+    [property: string]: unknown;
+}
+/**
+ * Outputs of inference for the Depth Estimation task
+ */
+export interface DepthEstimationOutput {
+    /**
+     * The predicted depth as an image
+     */
+    depth?: unknown;
+    /**
+     * The predicted depth as a tensor
+     */
+    predicted_depth?: unknown;
+    [property: string]: unknown;
+}
+//# sourceMappingURL=inference.d.ts.map

package/dist/tasks/depth-estimation/inference.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"inference.d.ts","sourceRoot":"","sources":["../../../src/tasks/depth-estimation/inference.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH;;GAEG;AACH,MAAM,WAAW,oBAAoB;IACpC;;OAEG;IACH,MAAM,EAAE,OAAO,CAAC;IAChB;;OAEG;IACH,UAAU,CAAC,EAAE;QAAE,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAA;KAAE,CAAC;IACxC,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B;AAED;;GAEG;AACH,MAAM,WAAW,qBAAqB;IACrC;;OAEG;IACH,KAAK,CAAC,EAAE,OAAO,CAAC;IAChB;;OAEG;IACH,eAAe,CAAC,EAAE,OAAO,CAAC;IAC1B,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B"}

package/dist/tasks/document-question-answering/data.d.cts

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

package/dist/tasks/document-question-answering/data.d.cts.map

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

package/dist/tasks/document-question-answering/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/tasks/document-question-answering/data.d.ts.map

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

package/dist/tasks/document-question-answering/inference.d.cts

@@ -0,0 +1,111 @@
+/**
+ * Inference code generated from the JSON schema spec in ./spec
+ *
+ * Using src/scripts/inference-codegen
+ */
+/**
+ * Inputs for Document Question Answering inference
+ */
+export interface DocumentQuestionAnsweringInput {
+    /**
+     * One (document, question) pair to answer
+     */
+    inputs: DocumentQuestionAnsweringInputData;
+    /**
+     * Additional inference parameters
+     */
+    parameters?: DocumentQuestionAnsweringParameters;
+    [property: string]: unknown;
+}
+/**
+ * One (document, question) pair to answer
+ */
+export interface DocumentQuestionAnsweringInputData {
+    /**
+     * The image on which the question is asked
+     */
+    image: unknown;
+    /**
+     * A question to ask of the document
+     */
+    question: string;
+    [property: string]: unknown;
+}
+/**
+ * Additional inference parameters
+ *
+ * Additional inference parameters for Document Question Answering
+ */
+export interface DocumentQuestionAnsweringParameters {
+    /**
+     * If the words in the document are too long to fit with the question for the model, it will
+     * be split in several chunks with some overlap. This argument controls the size of that
+     * overlap.
+     */
+    doc_stride?: number;
+    /**
+     * Whether to accept impossible as an answer
+     */
+    handle_impossible_answer?: boolean;
+    /**
+     * Language to use while running OCR. Defaults to english.
+     */
+    lang?: string;
+    /**
+     * The maximum length of predicted answers (e.g., only answers with a shorter length are
+     * considered).
+     */
+    max_answer_len?: number;
+    /**
+     * The maximum length of the question after tokenization. It will be truncated if needed.
+     */
+    max_question_len?: number;
+    /**
+     * The maximum length of the total sentence (context + question) in tokens of each chunk
+     * passed to the model. The context will be split in several chunks (using doc_stride as
+     * overlap) if needed.
+     */
+    max_seq_len?: number;
+    /**
+     * The number of answers to return (will be chosen by order of likelihood). Can return less
+     * than top_k answers if there are not enough options available within the context.
+     */
+    top_k?: number;
+    /**
+     * A list of words and bounding boxes (normalized 0->1000). If provided, the inference will
+     * skip the OCR step and use the provided bounding boxes instead.
+     */
+    word_boxes?: WordBox[];
+    [property: string]: unknown;
+}
+export type WordBox = number[] | string;
+export type DocumentQuestionAnsweringOutput = DocumentQuestionAnsweringOutputElement[];
+/**
+ * Outputs of inference for the Document Question Answering task
+ */
+export interface DocumentQuestionAnsweringOutputElement {
+    /**
+     * The answer to the question.
+     */
+    answer: string;
+    /**
+     * The end word index of the answer (in the OCR’d version of the input or provided word
+     * boxes).
+     */
+    end: number;
+    /**
+     * The probability associated to the answer.
+     */
+    score: number;
+    /**
+     * The start word index of the answer (in the OCR’d version of the input or provided word
+     * boxes).
+     */
+    start: number;
+    /**
+     * The index of each word/box pair that is in the answer
+     */
+    words: number[];
+    [property: string]: unknown;
+}
+//# sourceMappingURL=inference.d.cts.map

package/dist/tasks/document-question-answering/inference.d.cts.map

@@ -0,0 +1 @@
+{"version":3,"file":"inference.d.cts","sourceRoot":"","sources":["../../../src/tasks/document-question-answering/inference.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AACH;;GAEG;AACH,MAAM,WAAW,8BAA8B;IAC9C;;OAEG;IACH,MAAM,EAAE,kCAAkC,CAAC;IAC3C;;OAEG;IACH,UAAU,CAAC,EAAE,mCAAmC,CAAC;IACjD,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B;AACD;;GAEG;AACH,MAAM,WAAW,kCAAkC;IAClD;;OAEG;IACH,KAAK,EAAE,OAAO,CAAC;IACf;;OAEG;IACH,QAAQ,EAAE,MAAM,CAAC;IACjB,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B;AACD;;;;GAIG;AACH,MAAM,WAAW,mCAAmC;IACnD;;;;OAIG;IACH,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB;;OAEG;IACH,wBAAwB,CAAC,EAAE,OAAO,CAAC;IACnC;;OAEG;IACH,IAAI,CAAC,EAAE,MAAM,CAAC;IACd;;;OAGG;IACH,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB;;OAEG;IACH,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B;;;;OAIG;IACH,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB;;;OAGG;IACH,KAAK,CAAC,EAAE,MAAM,CAAC;IACf;;;OAGG;IACH,UAAU,CAAC,EAAE,OAAO,EAAE,CAAC;IACvB,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B;AACD,MAAM,MAAM,OAAO,GAAG,MAAM,EAAE,GAAG,MAAM,CAAC;AACxC,MAAM,MAAM,+BAA+B,GAAG,sCAAsC,EAAE,CAAC;AACvF;;GAEG;AACH,MAAM,WAAW,sCAAsC;IACtD;;OAEG;IACH,MAAM,EAAE,MAAM,CAAC;IACf;;;OAGG;IACH,GAAG,EAAE,MAAM,CAAC;IACZ;;OAEG;IACH,KAAK,EAAE,MAAM,CAAC;IACd;;;OAGG;IACH,KAAK,EAAE,MAAM,CAAC;IACd;;OAEG;IACH,KAAK,EAAE,MAAM,EAAE,CAAC;IAChB,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B"}

package/dist/tasks/document-question-answering/inference.d.ts

@@ -0,0 +1,111 @@
+/**
+ * Inference code generated from the JSON schema spec in ./spec
+ *
+ * Using src/scripts/inference-codegen
+ */
+/**
+ * Inputs for Document Question Answering inference
+ */
+export interface DocumentQuestionAnsweringInput {
+    /**
+     * One (document, question) pair to answer
+     */
+    inputs: DocumentQuestionAnsweringInputData;
+    /**
+     * Additional inference parameters
+     */
+    parameters?: DocumentQuestionAnsweringParameters;
+    [property: string]: unknown;
+}
+/**
+ * One (document, question) pair to answer
+ */
+export interface DocumentQuestionAnsweringInputData {
+    /**
+     * The image on which the question is asked
+     */
+    image: unknown;
+    /**
+     * A question to ask of the document
+     */
+    question: string;
+    [property: string]: unknown;
+}
+/**
+ * Additional inference parameters
+ *
+ * Additional inference parameters for Document Question Answering
+ */
+export interface DocumentQuestionAnsweringParameters {
+    /**
+     * If the words in the document are too long to fit with the question for the model, it will
+     * be split in several chunks with some overlap. This argument controls the size of that
+     * overlap.
+     */
+    doc_stride?: number;
+    /**
+     * Whether to accept impossible as an answer
+     */
+    handle_impossible_answer?: boolean;
+    /**
+     * Language to use while running OCR. Defaults to english.
+     */
+    lang?: string;
+    /**
+     * The maximum length of predicted answers (e.g., only answers with a shorter length are
+     * considered).
+     */
+    max_answer_len?: number;
+    /**
+     * The maximum length of the question after tokenization. It will be truncated if needed.
+     */
+    max_question_len?: number;
+    /**
+     * The maximum length of the total sentence (context + question) in tokens of each chunk
+     * passed to the model. The context will be split in several chunks (using doc_stride as
+     * overlap) if needed.
+     */
+    max_seq_len?: number;
+    /**
+     * The number of answers to return (will be chosen by order of likelihood). Can return less
+     * than top_k answers if there are not enough options available within the context.
+     */
+    top_k?: number;
+    /**
+     * A list of words and bounding boxes (normalized 0->1000). If provided, the inference will
+     * skip the OCR step and use the provided bounding boxes instead.
+     */
+    word_boxes?: WordBox[];
+    [property: string]: unknown;
+}
+export type WordBox = number[] | string;
+export type DocumentQuestionAnsweringOutput = DocumentQuestionAnsweringOutputElement[];
+/**
+ * Outputs of inference for the Document Question Answering task
+ */
+export interface DocumentQuestionAnsweringOutputElement {
+    /**
+     * The answer to the question.
+     */
+    answer: string;
+    /**
+     * The end word index of the answer (in the OCR’d version of the input or provided word
+     * boxes).
+     */
+    end: number;
+    /**
+     * The probability associated to the answer.
+     */
+    score: number;
+    /**
+     * The start word index of the answer (in the OCR’d version of the input or provided word
+     * boxes).
+     */
+    start: number;
+    /**
+     * The index of each word/box pair that is in the answer
+     */
+    words: number[];
+    [property: string]: unknown;
+}
+//# sourceMappingURL=inference.d.ts.map

package/dist/tasks/document-question-answering/inference.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"inference.d.ts","sourceRoot":"","sources":["../../../src/tasks/document-question-answering/inference.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AACH;;GAEG;AACH,MAAM,WAAW,8BAA8B;IAC9C;;OAEG;IACH,MAAM,EAAE,kCAAkC,CAAC;IAC3C;;OAEG;IACH,UAAU,CAAC,EAAE,mCAAmC,CAAC;IACjD,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B;AACD;;GAEG;AACH,MAAM,WAAW,kCAAkC;IAClD;;OAEG;IACH,KAAK,EAAE,OAAO,CAAC;IACf;;OAEG;IACH,QAAQ,EAAE,MAAM,CAAC;IACjB,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B;AACD;;;;GAIG;AACH,MAAM,WAAW,mCAAmC;IACnD;;;;OAIG;IACH,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB;;OAEG;IACH,wBAAwB,CAAC,EAAE,OAAO,CAAC;IACnC;;OAEG;IACH,IAAI,CAAC,EAAE,MAAM,CAAC;IACd;;;OAGG;IACH,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB;;OAEG;IACH,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B;;;;OAIG;IACH,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB;;;OAGG;IACH,KAAK,CAAC,EAAE,MAAM,CAAC;IACf;;;OAGG;IACH,UAAU,CAAC,EAAE,OAAO,EAAE,CAAC;IACvB,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B;AACD,MAAM,MAAM,OAAO,GAAG,MAAM,EAAE,GAAG,MAAM,CAAC;AACxC,MAAM,MAAM,+BAA+B,GAAG,sCAAsC,EAAE,CAAC;AACvF;;GAEG;AACH,MAAM,WAAW,sCAAsC;IACtD;;OAEG;IACH,MAAM,EAAE,MAAM,CAAC;IACf;;;OAGG;IACH,GAAG,EAAE,MAAM,CAAC;IACZ;;OAEG;IACH,KAAK,EAAE,MAAM,CAAC;IACd;;;OAGG;IACH,KAAK,EAAE,MAAM,CAAC;IACd;;OAEG;IACH,KAAK,EAAE,MAAM,EAAE,CAAC;IAChB,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B"}

package/dist/tasks/feature-extraction/data.d.cts

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

package/dist/tasks/feature-extraction/data.d.cts.map

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

package/dist/tasks/feature-extraction/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/tasks/feature-extraction/data.d.ts.map

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

package/dist/tasks/feature-extraction/inference.d.cts

@@ -0,0 +1,38 @@
+/**
+ * Inference code generated from the JSON schema spec in ./spec
+ *
+ * Using src/scripts/inference-codegen
+ */
+export type FeatureExtractionOutput = Array<number[]>;
+/**
+ * Feature Extraction Input.
+ *
+ * Auto-generated from TEI specs.
+ * For more details, check out
+ * https://github.com/huggingface/huggingface.js/blob/main/packages/tasks/scripts/inference-tei-import.ts.
+ */
+export interface FeatureExtractionInput {
+    /**
+     * The text to embed.
+     */
+    inputs: string;
+    normalize?: boolean;
+    /**
+     * The name of the prompt that should be used by for encoding. If not set, no prompt
+     * will be applied.
+     *
+     * Must be a key in the `Sentence Transformers` configuration `prompts` dictionary.
+     *
+     * For example if ``prompt_name`` is "query" and the ``prompts`` is {"query": "query: ",
+     * ...},
+     * then the sentence "What is the capital of France?" will be encoded as
+     * "query: What is the capital of France?" because the prompt text will be prepended before
+     * any text to encode.
+     */
+    prompt_name?: string;
+    truncate?: boolean;
+    truncation_direction?: FeatureExtractionInputTruncationDirection;
+    [property: string]: unknown;
+}
+export type FeatureExtractionInputTruncationDirection = "Left" | "Right";
+//# sourceMappingURL=inference.d.cts.map

package/dist/tasks/feature-extraction/inference.d.cts.map

@@ -0,0 +1 @@
+{"version":3,"file":"inference.d.cts","sourceRoot":"","sources":["../../../src/tasks/feature-extraction/inference.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,MAAM,MAAM,uBAAuB,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC;AAEtD;;;;;;GAMG;AACH,MAAM,WAAW,sBAAsB;IACtC;;OAEG;IACH,MAAM,EAAE,MAAM,CAAC;IACf,SAAS,CAAC,EAAE,OAAO,CAAC;IACpB;;;;;;;;;;;OAWG;IACH,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,QAAQ,CAAC,EAAE,OAAO,CAAC;IACnB,oBAAoB,CAAC,EAAE,yCAAyC,CAAC;IACjE,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B;AAED,MAAM,MAAM,yCAAyC,GAAG,MAAM,GAAG,OAAO,CAAC"}

package/dist/tasks/feature-extraction/inference.d.ts

@@ -0,0 +1,38 @@
+/**
+ * Inference code generated from the JSON schema spec in ./spec
+ *
+ * Using src/scripts/inference-codegen
+ */
+export type FeatureExtractionOutput = Array<number[]>;
+/**
+ * Feature Extraction Input.
+ *
+ * Auto-generated from TEI specs.
+ * For more details, check out
+ * https://github.com/huggingface/huggingface.js/blob/main/packages/tasks/scripts/inference-tei-import.ts.
+ */
+export interface FeatureExtractionInput {
+    /**
+     * The text to embed.
+     */
+    inputs: string;
+    normalize?: boolean;
+    /**
+     * The name of the prompt that should be used by for encoding. If not set, no prompt
+     * will be applied.
+     *
+     * Must be a key in the `Sentence Transformers` configuration `prompts` dictionary.
+     *
+     * For example if ``prompt_name`` is "query" and the ``prompts`` is {"query": "query: ",
+     * ...},
+     * then the sentence "What is the capital of France?" will be encoded as
+     * "query: What is the capital of France?" because the prompt text will be prepended before
+     * any text to encode.
+     */
+    prompt_name?: string;
+    truncate?: boolean;
+    truncation_direction?: FeatureExtractionInputTruncationDirection;
+    [property: string]: unknown;
+}
+export type FeatureExtractionInputTruncationDirection = "Left" | "Right";
+//# sourceMappingURL=inference.d.ts.map

package/dist/tasks/feature-extraction/inference.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"inference.d.ts","sourceRoot":"","sources":["../../../src/tasks/feature-extraction/inference.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,MAAM,MAAM,uBAAuB,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC;AAEtD;;;;;;GAMG;AACH,MAAM,WAAW,sBAAsB;IACtC;;OAEG;IACH,MAAM,EAAE,MAAM,CAAC;IACf,SAAS,CAAC,EAAE,OAAO,CAAC;IACpB;;;;;;;;;;;OAWG;IACH,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,QAAQ,CAAC,EAAE,OAAO,CAAC;IACnB,oBAAoB,CAAC,EAAE,yCAAyC,CAAC;IACjE,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B;AAED,MAAM,MAAM,yCAAyC,GAAG,MAAM,GAAG,OAAO,CAAC"}

package/dist/tasks/fill-mask/data.d.cts

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

package/dist/tasks/fill-mask/data.d.cts.map

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

package/dist/tasks/fill-mask/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/tasks/fill-mask/data.d.ts.map

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

package/dist/tasks/fill-mask/inference.d.cts

@@ -0,0 +1,63 @@
+/**
+ * Inference code generated from the JSON schema spec in ./spec
+ *
+ * Using src/scripts/inference-codegen
+ */
+/**
+ * Inputs for Fill Mask inference
+ */
+export interface FillMaskInput {
+    /**
+     * The text with masked tokens
+     */
+    inputs: string;
+    /**
+     * Additional inference parameters
+     */
+    parameters?: FillMaskParameters;
+    [property: string]: unknown;
+}
+/**
+ * Additional inference parameters
+ *
+ * Additional inference parameters for Fill Mask
+ */
+export interface FillMaskParameters {
+    /**
+     * When passed, the model will limit the scores to the passed targets instead of looking up
+     * in the whole vocabulary. If the provided targets are not in the model vocab, they will be
+     * tokenized and the first resulting token will be used (with a warning, and that might be
+     * slower).
+     */
+    targets?: string[];
+    /**
+     * When passed, overrides the number of predictions to return.
+     */
+    top_k?: number;
+    [property: string]: unknown;
+}
+export type FillMaskOutput = FillMaskOutputElement[];
+/**
+ * Outputs of inference for the Fill Mask task
+ */
+export interface FillMaskOutputElement {
+    /**
+     * The corresponding probability
+     */
+    score: number;
+    /**
+     * The corresponding input with the mask token prediction.
+     */
+    sequence: string;
+    /**
+     * The predicted token id (to replace the masked one).
+     */
+    token: number;
+    tokenStr: unknown;
+    /**
+     * The predicted token (to replace the masked one).
+     */
+    token_str?: string;
+    [property: string]: unknown;
+}
+//# sourceMappingURL=inference.d.cts.map

package/dist/tasks/fill-mask/inference.d.cts.map

@@ -0,0 +1 @@
+{"version":3,"file":"inference.d.cts","sourceRoot":"","sources":["../../../src/tasks/fill-mask/inference.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AACH;;GAEG;AACH,MAAM,WAAW,aAAa;IAC7B;;OAEG;IACH,MAAM,EAAE,MAAM,CAAC;IACf;;OAEG;IACH,UAAU,CAAC,EAAE,kBAAkB,CAAC;IAChC,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B;AACD;;;;GAIG;AACH,MAAM,WAAW,kBAAkB;IAClC;;;;;OAKG;IACH,OAAO,CAAC,EAAE,MAAM,EAAE,CAAC;IACnB;;OAEG;IACH,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B;AACD,MAAM,MAAM,cAAc,GAAG,qBAAqB,EAAE,CAAC;AACrD;;GAEG;AACH,MAAM,WAAW,qBAAqB;IACrC;;OAEG;IACH,KAAK,EAAE,MAAM,CAAC;IACd;;OAEG;IACH,QAAQ,EAAE,MAAM,CAAC;IACjB;;OAEG;IACH,KAAK,EAAE,MAAM,CAAC;IACd,QAAQ,EAAE,OAAO,CAAC;IAClB;;OAEG;IACH,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B"}

package/dist/tasks/fill-mask/inference.d.ts

@@ -0,0 +1,63 @@
+/**
+ * Inference code generated from the JSON schema spec in ./spec
+ *
+ * Using src/scripts/inference-codegen
+ */
+/**
+ * Inputs for Fill Mask inference
+ */
+export interface FillMaskInput {
+    /**
+     * The text with masked tokens
+     */
+    inputs: string;
+    /**
+     * Additional inference parameters
+     */
+    parameters?: FillMaskParameters;
+    [property: string]: unknown;
+}
+/**
+ * Additional inference parameters
+ *
+ * Additional inference parameters for Fill Mask
+ */
+export interface FillMaskParameters {
+    /**
+     * When passed, the model will limit the scores to the passed targets instead of looking up
+     * in the whole vocabulary. If the provided targets are not in the model vocab, they will be
+     * tokenized and the first resulting token will be used (with a warning, and that might be
+     * slower).
+     */
+    targets?: string[];
+    /**
+     * When passed, overrides the number of predictions to return.
+     */
+    top_k?: number;
+    [property: string]: unknown;
+}
+export type FillMaskOutput = FillMaskOutputElement[];
+/**
+ * Outputs of inference for the Fill Mask task
+ */
+export interface FillMaskOutputElement {
+    /**
+     * The corresponding probability
+     */
+    score: number;
+    /**
+     * The corresponding input with the mask token prediction.
+     */
+    sequence: string;
+    /**
+     * The predicted token id (to replace the masked one).
+     */
+    token: number;
+    tokenStr: unknown;
+    /**
+     * The predicted token (to replace the masked one).
+     */
+    token_str?: string;
+    [property: string]: unknown;
+}
+//# sourceMappingURL=inference.d.ts.map

package/dist/tasks/fill-mask/inference.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"inference.d.ts","sourceRoot":"","sources":["../../../src/tasks/fill-mask/inference.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AACH;;GAEG;AACH,MAAM,WAAW,aAAa;IAC7B;;OAEG;IACH,MAAM,EAAE,MAAM,CAAC;IACf;;OAEG;IACH,UAAU,CAAC,EAAE,kBAAkB,CAAC;IAChC,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B;AACD;;;;GAIG;AACH,MAAM,WAAW,kBAAkB;IAClC;;;;;OAKG;IACH,OAAO,CAAC,EAAE,MAAM,EAAE,CAAC;IACnB;;OAEG;IACH,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B;AACD,MAAM,MAAM,cAAc,GAAG,qBAAqB,EAAE,CAAC;AACrD;;GAEG;AACH,MAAM,WAAW,qBAAqB;IACrC;;OAEG;IACH,KAAK,EAAE,MAAM,CAAC;IACd;;OAEG;IACH,QAAQ,EAAE,MAAM,CAAC;IACjB;;OAEG;IACH,KAAK,EAAE,MAAM,CAAC;IACd,QAAQ,EAAE,OAAO,CAAC;IAClB;;OAEG;IACH,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC5B"}

package/dist/tasks/image-classification/data.d.cts

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

package/dist/tasks/image-classification/data.d.cts.map

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

package/dist/tasks/image-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/tasks/image-classification/data.d.ts.map

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

package/dist/tasks/image-classification/inference.d.cts

@@ -0,0 +1,53 @@
+/**
+ * Inference code generated from the JSON schema spec in ./spec
+ *
+ * Using src/scripts/inference-codegen
+ */
+/**
+ * Inputs for Image Classification inference
+ */
+export interface ImageClassificationInput {
+    /**
+     * The input image data as a base64-encoded string. If no `parameters` are provided, you can
+     * also provide the image data as a raw bytes payload.
+     */
+    inputs: string;
+    /**
+     * Additional inference parameters
+     */
+    parameters?: ImageClassificationParameters;
+    [property: string]: unknown;
+}
+/**
+ * Additional inference parameters
+ *
+ * Additional inference parameters for Image Classification
+ */
+export interface ImageClassificationParameters {
+    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 ImageClassificationOutput = ImageClassificationOutputElement[];
+/**
+ * Outputs of inference for the Image Classification task
+ */
+export interface ImageClassificationOutputElement {
+    /**
+     * The predicted class label.
+     */
+    label: string;
+    /**
+     * The corresponding probability.
+     */
+    score: number;
+    [property: string]: unknown;
+}
+//# sourceMappingURL=inference.d.cts.map

package/dist/tasks/image-classification/inference.d.cts.map

@@ -0,0 +1 @@
+{"version":3,"file":"inference.d.cts","sourceRoot":"","sources":["../../../src/tasks/image-classification/inference.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AACH;;GAEG;AACH,MAAM,WAAW,wBAAwB;IACxC;;;OAGG;IACH,MAAM,EAAE,MAAM,CAAC;IACf;;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"}