@huggingface/inference 3.0.1 → 3.1.0

package/dist/src/tasks/cv/imageToImage.d.ts

@@ -1,55 +1,9 @@
+import type { ImageToImageInput } from "@huggingface/tasks";
 import type { BaseArgs, Options } from "../../types";
-export type ImageToImageArgs = BaseArgs & {
-    /**
-     * The initial image condition
-     *
-     **/
-    inputs: Blob | ArrayBuffer;
-    parameters?: {
-        /**
-         * The text prompt to guide the image generation.
-         */
-        prompt?: string;
-        /**
-         * strengh param only works for SD img2img and alt diffusion img2img models
-         * Conceptually, indicates how much to transform the reference `image`. Must be between 0 and 1. `image`
-         * will be used as a starting point, adding more noise to it the larger the `strength`. The number of
-         * denoising steps depends on the amount of noise initially added. When `strength` is 1, added noise will
-         * be maximum and the denoising process will run for the full number of iterations specified in
-         * `num_inference_steps`. A value of 1, therefore, essentially ignores `image`.
-         **/
-        strength?: number;
-        /**
-         * An optional negative prompt for the image generation
-         */
-        negative_prompt?: string;
-        /**
-         * The height in pixels of the generated image
-         */
-        height?: number;
-        /**
-         * The width in pixels of the generated image
-         */
-        width?: number;
-        /**
-         * The number of denoising steps. More denoising steps usually lead to a higher quality image at the expense of slower inference.
-         */
-        num_inference_steps?: number;
-        /**
-         * Guidance scale: Higher guidance scale encourages to generate images that are closely linked to the text `prompt`, usually at the expense of lower image quality.
-         */
-        guidance_scale?: number;
-        /**
-         * guess_mode only works for ControlNet models, defaults to False In this mode, the ControlNet encoder will try best to recognize the content of the input image even if
-         * you remove all prompts. The `guidance_scale` between 3.0 and 5.0 is recommended.
-         */
-        guess_mode?: boolean;
-    };
-};
-export type ImageToImageOutput = Blob;
+export type ImageToImageArgs = BaseArgs & ImageToImageInput;
 /**
  * This task reads some text input and outputs an image.
  * Recommended model: lllyasviel/sd-controlnet-depth
  */
-export declare function imageToImage(args: ImageToImageArgs, options?: Options): Promise<ImageToImageOutput>;
+export declare function imageToImage(args: ImageToImageArgs, options?: Options): Promise<Blob>;
 //# sourceMappingURL=imageToImage.d.ts.map

package/dist/src/tasks/cv/imageToImage.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"imageToImage.d.ts","sourceRoot":"","sources":["../../../../src/tasks/cv/imageToImage.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAe,MAAM,aAAa,CAAC;AAIlE,MAAM,MAAM,gBAAgB,GAAG,QAAQ,GAAG;IACzC;;;QAGI;IACJ,MAAM,EAAE,IAAI,GAAG,WAAW,CAAC;IAE3B,UAAU,CAAC,EAAE;QACZ;;WAEG;QACH,MAAM,CAAC,EAAE,MAAM,CAAC;QAChB;;;;;;;YAOI;QACJ,QAAQ,CAAC,EAAE,MAAM,CAAC;QAClB;;WAEG;QACH,eAAe,CAAC,EAAE,MAAM,CAAC;QACzB;;WAEG;QACH,MAAM,CAAC,EAAE,MAAM,CAAC;QAChB;;WAEG;QACH,KAAK,CAAC,EAAE,MAAM,CAAC;QACf;;WAEG;QACH,mBAAmB,CAAC,EAAE,MAAM,CAAC;QAC7B;;WAEG;QACH,cAAc,CAAC,EAAE,MAAM,CAAC;QACxB;;;WAGG;QACH,UAAU,CAAC,EAAE,OAAO,CAAC;KACrB,CAAC;CACF,CAAC;AAEF,MAAM,MAAM,kBAAkB,GAAG,IAAI,CAAC;AAEtC;;;GAGG;AACH,wBAAsB,YAAY,CAAC,IAAI,EAAE,gBAAgB,EAAE,OAAO,CAAC,EAAE,OAAO,GAAG,OAAO,CAAC,kBAAkB,CAAC,CAyBzG"}
+{"version":3,"file":"imageToImage.d.ts","sourceRoot":"","sources":["../../../../src/tasks/cv/imageToImage.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,oBAAoB,CAAC;AAE5D,OAAO,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAe,MAAM,aAAa,CAAC;AAIlE,MAAM,MAAM,gBAAgB,GAAG,QAAQ,GAAG,iBAAiB,CAAC;AAE5D;;;GAGG;AACH,wBAAsB,YAAY,CAAC,IAAI,EAAE,gBAAgB,EAAE,OAAO,CAAC,EAAE,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC,CAyB3F"}

package/dist/src/tasks/cv/imageToText.d.ts

@@ -1,16 +1,7 @@
+import type { ImageToTextInput, ImageToTextOutput } from "@huggingface/tasks";
 import type { BaseArgs, Options } from "../../types";
-export type ImageToTextArgs = BaseArgs & {
-    /**
-     * Binary image data
-     */
-    data: Blob | ArrayBuffer;
-};
-export interface ImageToTextOutput {
-    /**
-     * The generated caption
-     */
-    generated_text: string;
-}
+import type { LegacyImageInput } from "./utils";
+export type ImageToTextArgs = BaseArgs & (ImageToTextInput | LegacyImageInput);
 /**
  * This task reads some image input and outputs the text caption.
  */

package/dist/src/tasks/cv/imageToText.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"imageToText.d.ts","sourceRoot":"","sources":["../../../../src/tasks/cv/imageToText.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,aAAa,CAAC;AAGrD,MAAM,MAAM,eAAe,GAAG,QAAQ,GAAG;IACxC;;OAEG;IACH,IAAI,EAAE,IAAI,GAAG,WAAW,CAAC;CACzB,CAAC;AAEF,MAAM,WAAW,iBAAiB;IACjC;;OAEG;IACH,cAAc,EAAE,MAAM,CAAC;CACvB;AAED;;GAEG;AACH,wBAAsB,WAAW,CAAC,IAAI,EAAE,eAAe,EAAE,OAAO,CAAC,EAAE,OAAO,GAAG,OAAO,CAAC,iBAAiB,CAAC,CAatG"}
+{"version":3,"file":"imageToText.d.ts","sourceRoot":"","sources":["../../../../src/tasks/cv/imageToText.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,gBAAgB,EAAE,iBAAiB,EAAE,MAAM,oBAAoB,CAAC;AAE9E,OAAO,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,aAAa,CAAC;AAErD,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,SAAS,CAAC;AAGhD,MAAM,MAAM,eAAe,GAAG,QAAQ,GAAG,CAAC,gBAAgB,GAAG,gBAAgB,CAAC,CAAC;AAC/E;;GAEG;AACH,wBAAsB,WAAW,CAAC,IAAI,EAAE,eAAe,EAAE,OAAO,CAAC,EAAE,OAAO,GAAG,OAAO,CAAC,iBAAiB,CAAC,CActG"}

package/dist/src/tasks/cv/objectDetection.d.ts

@@ -1,30 +1,7 @@
 import type { BaseArgs, Options } from "../../types";
-export type ObjectDetectionArgs = BaseArgs & {
-    /**
-     * Binary image data
-     */
-    data: Blob | ArrayBuffer;
-};
-export interface ObjectDetectionOutputValue {
-    /**
-     * A dict (with keys [xmin,ymin,xmax,ymax]) representing the bounding box of a detected object.
-     */
-    box: {
-        xmax: number;
-        xmin: number;
-        ymax: number;
-        ymin: number;
-    };
-    /**
-     * The label for the class (model specific) of a detected object.
-     */
-    label: string;
-    /**
-     * A float that represents how likely it is that the detected object belongs to the given class.
-     */
-    score: number;
-}
-export type ObjectDetectionOutput = ObjectDetectionOutputValue[];
+import type { ObjectDetectionInput, ObjectDetectionOutput } from "@huggingface/tasks";
+import { type LegacyImageInput } from "./utils";
+export type ObjectDetectionArgs = BaseArgs & (ObjectDetectionInput | LegacyImageInput);
 /**
  * This task reads some image input and outputs the likelihood of classes & bounding boxes of detected objects.
  * Recommended model: facebook/detr-resnet-50

package/dist/src/tasks/cv/objectDetection.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"objectDetection.d.ts","sourceRoot":"","sources":["../../../../src/tasks/cv/objectDetection.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,aAAa,CAAC;AAGrD,MAAM,MAAM,mBAAmB,GAAG,QAAQ,GAAG;IAC5C;;OAEG;IACH,IAAI,EAAE,IAAI,GAAG,WAAW,CAAC;CACzB,CAAC;AAEF,MAAM,WAAW,0BAA0B;IAC1C;;OAEG;IACH,GAAG,EAAE;QACJ,IAAI,EAAE,MAAM,CAAC;QACb,IAAI,EAAE,MAAM,CAAC;QACb,IAAI,EAAE,MAAM,CAAC;QACb,IAAI,EAAE,MAAM,CAAC;KACb,CAAC;IACF;;OAEG;IACH,KAAK,EAAE,MAAM,CAAC;IAEd;;OAEG;IACH,KAAK,EAAE,MAAM,CAAC;CACd;AAED,MAAM,MAAM,qBAAqB,GAAG,0BAA0B,EAAE,CAAC;AAEjE;;;GAGG;AACH,wBAAsB,eAAe,CAAC,IAAI,EAAE,mBAAmB,EAAE,OAAO,CAAC,EAAE,OAAO,GAAG,OAAO,CAAC,qBAAqB,CAAC,CAsBlH"}
+{"version":3,"file":"objectDetection.d.ts","sourceRoot":"","sources":["../../../../src/tasks/cv/objectDetection.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,aAAa,CAAC;AAErD,OAAO,KAAK,EAAE,oBAAoB,EAAE,qBAAqB,EAAE,MAAM,oBAAoB,CAAC;AACtF,OAAO,EAAkB,KAAK,gBAAgB,EAAE,MAAM,SAAS,CAAC;AAEhE,MAAM,MAAM,mBAAmB,GAAG,QAAQ,GAAG,CAAC,oBAAoB,GAAG,gBAAgB,CAAC,CAAC;AAEvF;;;GAGG;AACH,wBAAsB,eAAe,CAAC,IAAI,EAAE,mBAAmB,EAAE,OAAO,CAAC,EAAE,OAAO,GAAG,OAAO,CAAC,qBAAqB,CAAC,CAuBlH"}

package/dist/src/tasks/cv/textToImage.d.ts

@@ -1,44 +1,9 @@
+import type { TextToImageInput } from "@huggingface/tasks";
 import type { BaseArgs, Options } from "../../types";
-export type TextToImageArgs = BaseArgs & {
-    /**
-     * The text to generate an image from
-     */
-    inputs: string;
-    /**
-     * Same param but for external providers like Together, Replicate
-     */
-    prompt?: string;
-    response_format?: "base64";
-    input?: {
-        prompt: string;
-    };
-    parameters?: {
-        /**
-         * An optional negative prompt for the image generation
-         */
-        negative_prompt?: string;
-        /**
-         * The height in pixels of the generated image
-         */
-        height?: number;
-        /**
-         * The width in pixels of the generated image
-         */
-        width?: number;
-        /**
-         * The number of denoising steps. More denoising steps usually lead to a higher quality image at the expense of slower inference.
-         */
-        num_inference_steps?: number;
-        /**
-         * Guidance scale: Higher guidance scale encourages to generate images that are closely linked to the text `prompt`, usually at the expense of lower image quality.
-         */
-        guidance_scale?: number;
-    };
-};
-export type TextToImageOutput = Blob;
+export type TextToImageArgs = BaseArgs & TextToImageInput;
 /**
  * This task reads some text input and outputs an image.
  * Recommended model: stabilityai/stable-diffusion-2
  */
-export declare function textToImage(args: TextToImageArgs, options?: Options): Promise<TextToImageOutput>;
+export declare function textToImage(args: TextToImageArgs, options?: Options): Promise<Blob>;
 //# sourceMappingURL=textToImage.d.ts.map

package/dist/src/tasks/cv/textToImage.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"textToImage.d.ts","sourceRoot":"","sources":["../../../../src/tasks/cv/textToImage.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,aAAa,CAAC;AAGrD,MAAM,MAAM,eAAe,GAAG,QAAQ,GAAG;IACxC;;OAEG;IACH,MAAM,EAAE,MAAM,CAAC;IAEf;;OAEG;IACH,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,eAAe,CAAC,EAAE,QAAQ,CAAC;IAC3B,KAAK,CAAC,EAAE;QACP,MAAM,EAAE,MAAM,CAAC;KACf,CAAC;IAEF,UAAU,CAAC,EAAE;QACZ;;WAEG;QACH,eAAe,CAAC,EAAE,MAAM,CAAC;QACzB;;WAEG;QACH,MAAM,CAAC,EAAE,MAAM,CAAC;QAChB;;WAEG;QACH,KAAK,CAAC,EAAE,MAAM,CAAC;QACf;;WAEG;QACH,mBAAmB,CAAC,EAAE,MAAM,CAAC;QAC7B;;WAEG;QACH,cAAc,CAAC,EAAE,MAAM,CAAC;KACxB,CAAC;CACF,CAAC;AAEF,MAAM,MAAM,iBAAiB,GAAG,IAAI,CAAC;AAWrC;;;GAGG;AACH,wBAAsB,WAAW,CAAC,IAAI,EAAE,eAAe,EAAE,OAAO,CAAC,EAAE,OAAO,GAAG,OAAO,CAAC,iBAAiB,CAAC,CAmCtG"}
+{"version":3,"file":"textToImage.d.ts","sourceRoot":"","sources":["../../../../src/tasks/cv/textToImage.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,gBAAgB,EAAqB,MAAM,oBAAoB,CAAC;AAE9E,OAAO,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,aAAa,CAAC;AAIrD,MAAM,MAAM,eAAe,GAAG,QAAQ,GAAG,gBAAgB,CAAC;AAW1D;;;GAGG;AACH,wBAAsB,WAAW,CAAC,IAAI,EAAE,eAAe,EAAE,OAAO,CAAC,EAAE,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC,CAoCzF"}

package/dist/src/tasks/cv/textToVideo.d.ts

@@ -0,0 +1,6 @@
+import type { BaseArgs, Options } from "../../types";
+import type { TextToVideoInput } from "@huggingface/tasks";
+export type TextToVideoArgs = BaseArgs & TextToVideoInput;
+export type TextToVideoOutput = Blob;
+export declare function textToVideo(args: TextToVideoArgs, options?: Options): Promise<TextToVideoOutput>;
+//# sourceMappingURL=textToVideo.d.ts.map

package/dist/src/tasks/cv/textToVideo.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"textToVideo.d.ts","sourceRoot":"","sources":["../../../../src/tasks/cv/textToVideo.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,QAAQ,EAAqB,OAAO,EAAE,MAAM,aAAa,CAAC;AACxE,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,oBAAoB,CAAC;AAO3D,MAAM,MAAM,eAAe,GAAG,QAAQ,GAAG,gBAAgB,CAAC;AAE1D,MAAM,MAAM,iBAAiB,GAAG,IAAI,CAAC;AAcrC,wBAAsB,WAAW,CAAC,IAAI,EAAE,eAAe,EAAE,OAAO,CAAC,EAAE,OAAO,GAAG,OAAO,CAAC,iBAAiB,CAAC,CA0CtG"}

package/dist/src/tasks/cv/utils.d.ts

@@ -0,0 +1,11 @@
+import type { BaseArgs, RequestArgs } from "../../types";
+/**
+ * @deprecated
+ */
+export interface LegacyImageInput {
+    data: Blob | ArrayBuffer;
+}
+export declare function preparePayload(args: BaseArgs & ({
+    inputs: Blob;
+} | LegacyImageInput)): RequestArgs;
+//# sourceMappingURL=utils.d.ts.map

package/dist/src/tasks/cv/utils.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"utils.d.ts","sourceRoot":"","sources":["../../../../src/tasks/cv/utils.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,QAAQ,EAAE,WAAW,EAAE,MAAM,aAAa,CAAC;AAGzD;;GAEG;AACH,MAAM,WAAW,gBAAgB;IAChC,IAAI,EAAE,IAAI,GAAG,WAAW,CAAC;CACzB;AAED,wBAAgB,cAAc,CAAC,IAAI,EAAE,QAAQ,GAAG,CAAC;IAAE,MAAM,EAAE,IAAI,CAAA;CAAE,GAAG,gBAAgB,CAAC,GAAG,WAAW,CAElG"}

package/dist/src/tasks/cv/zeroShotImageClassification.d.ts

@@ -1,26 +1,18 @@
 import type { BaseArgs, Options } from "../../types";
-export type ZeroShotImageClassificationArgs = BaseArgs & {
+import type { ZeroShotImageClassificationInput, ZeroShotImageClassificationOutput } from "@huggingface/tasks";
+/**
+ * @deprecated
+ */
+interface LegacyZeroShotImageClassificationInput {
     inputs: {
-        /**
-         * Binary image data
-         */
         image: Blob | ArrayBuffer;
     };
-    parameters: {
-        /**
-         * A list of strings that are potential classes for inputs. (max 10)
-         */
-        candidate_labels: string[];
-    };
-};
-export interface ZeroShotImageClassificationOutputValue {
-    label: string;
-    score: number;
 }
-export type ZeroShotImageClassificationOutput = ZeroShotImageClassificationOutputValue[];
+export type ZeroShotImageClassificationArgs = BaseArgs & (ZeroShotImageClassificationInput | LegacyZeroShotImageClassificationInput);
 /**
  * Classify an image to specified classes.
  * Recommended model: openai/clip-vit-large-patch14-336
  */
 export declare function zeroShotImageClassification(args: ZeroShotImageClassificationArgs, options?: Options): Promise<ZeroShotImageClassificationOutput>;
+export {};
 //# sourceMappingURL=zeroShotImageClassification.d.ts.map

package/dist/src/tasks/cv/zeroShotImageClassification.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"zeroShotImageClassification.d.ts","sourceRoot":"","sources":["../../../../src/tasks/cv/zeroShotImageClassification.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,aAAa,CAAC;AAKrD,MAAM,MAAM,+BAA+B,GAAG,QAAQ,GAAG;IACxD,MAAM,EAAE;QACP;;WAEG;QACH,KAAK,EAAE,IAAI,GAAG,WAAW,CAAC;KAC1B,CAAC;IACF,UAAU,EAAE;QACX;;WAEG;QACH,gBAAgB,EAAE,MAAM,EAAE,CAAC;KAC3B,CAAC;CACF,CAAC;AAEF,MAAM,WAAW,sCAAsC;IACtD,KAAK,EAAE,MAAM,CAAC;IACd,KAAK,EAAE,MAAM,CAAC;CACd;AAED,MAAM,MAAM,iCAAiC,GAAG,sCAAsC,EAAE,CAAC;AAEzF;;;GAGG;AACH,wBAAsB,2BAA2B,CAChD,IAAI,EAAE,+BAA+B,EACrC,OAAO,CAAC,EAAE,OAAO,GACf,OAAO,CAAC,iCAAiC,CAAC,CAsB5C"}
+{"version":3,"file":"zeroShotImageClassification.d.ts","sourceRoot":"","sources":["../../../../src/tasks/cv/zeroShotImageClassification.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,aAAa,CAAC;AAIrD,OAAO,KAAK,EAAE,gCAAgC,EAAE,iCAAiC,EAAE,MAAM,oBAAoB,CAAC;AAE9G;;GAEG;AACH,UAAU,sCAAsC;IAC/C,MAAM,EAAE;QAAE,KAAK,EAAE,IAAI,GAAG,WAAW,CAAA;KAAE,CAAC;CACtC;AAED,MAAM,MAAM,+BAA+B,GAAG,QAAQ,GACrD,CAAC,gCAAgC,GAAG,sCAAsC,CAAC,CAAC;AAwB7E;;;GAGG;AACH,wBAAsB,2BAA2B,CAChD,IAAI,EAAE,+BAA+B,EACrC,OAAO,CAAC,EAAE,OAAO,GACf,OAAO,CAAC,iCAAiC,CAAC,CAY5C"}

package/dist/src/tasks/multimodal/documentQuestionAnswering.d.ts

@@ -1,35 +1,12 @@
 import type { BaseArgs, Options } from "../../types";
-export type DocumentQuestionAnsweringArgs = BaseArgs & {
-    inputs: {
-        /**
-         * Raw image
-         *
-         * You can use native `File` in browsers, or `new Blob([buffer])` in node, or for a base64 image `new Blob([btoa(base64String)])`, or even `await (await fetch('...)).blob()`
-         **/
-        image: Blob | ArrayBuffer;
-        question: string;
+import type { DocumentQuestionAnsweringInput, DocumentQuestionAnsweringInputData, DocumentQuestionAnsweringOutput } from "@huggingface/tasks";
+export type DocumentQuestionAnsweringArgs = BaseArgs & DocumentQuestionAnsweringInput & {
+    inputs: DocumentQuestionAnsweringInputData & {
+        image: Blob;
     };
 };
-export interface DocumentQuestionAnsweringOutput {
-    /**
-     * A string that’s the answer within the document.
-     */
-    answer: string;
-    /**
-     * ?
-     */
-    end?: number;
-    /**
-     * A float that represents how likely that the answer is correct
-     */
-    score?: number;
-    /**
-     * ?
-     */
-    start?: number;
-}
 /**
  * Answers a question on a document image. Recommended model: impira/layoutlm-document-qa.
  */
-export declare function documentQuestionAnswering(args: DocumentQuestionAnsweringArgs, options?: Options): Promise<DocumentQuestionAnsweringOutput>;
+export declare function documentQuestionAnswering(args: DocumentQuestionAnsweringArgs, options?: Options): Promise<DocumentQuestionAnsweringOutput[number]>;
 //# sourceMappingURL=documentQuestionAnswering.d.ts.map

package/dist/src/tasks/multimodal/documentQuestionAnswering.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"documentQuestionAnswering.d.ts","sourceRoot":"","sources":["../../../../src/tasks/multimodal/documentQuestionAnswering.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,aAAa,CAAC;AAMrD,MAAM,MAAM,6BAA6B,GAAG,QAAQ,GAAG;IACtD,MAAM,EAAE;QACP;;;;YAII;QACJ,KAAK,EAAE,IAAI,GAAG,WAAW,CAAC;QAC1B,QAAQ,EAAE,MAAM,CAAC;KACjB,CAAC;CACF,CAAC;AAEF,MAAM,WAAW,+BAA+B;IAC/C;;OAEG;IACH,MAAM,EAAE,MAAM,CAAC;IACf;;OAEG;IACH,GAAG,CAAC,EAAE,MAAM,CAAC;IACb;;OAEG;IACH,KAAK,CAAC,EAAE,MAAM,CAAC;IACf;;OAEG;IACH,KAAK,CAAC,EAAE,MAAM,CAAC;CACf;AAED;;GAEG;AACH,wBAAsB,yBAAyB,CAC9C,IAAI,EAAE,6BAA6B,EACnC,OAAO,CAAC,EAAE,OAAO,GACf,OAAO,CAAC,+BAA+B,CAAC,CA4B1C"}
+{"version":3,"file":"documentQuestionAnswering.d.ts","sourceRoot":"","sources":["../../../../src/tasks/multimodal/documentQuestionAnswering.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,aAAa,CAAC;AAKrD,OAAO,KAAK,EACX,8BAA8B,EAC9B,kCAAkC,EAClC,+BAA+B,EAC/B,MAAM,oBAAoB,CAAC;AAG5B,MAAM,MAAM,6BAA6B,GAAG,QAAQ,GACnD,8BAA8B,GAAG;IAAE,MAAM,EAAE,kCAAkC,GAAG;QAAE,KAAK,EAAE,IAAI,CAAA;KAAE,CAAA;CAAE,CAAC;AAEnG;;GAEG;AACH,wBAAsB,yBAAyB,CAC9C,IAAI,EAAE,6BAA6B,EACnC,OAAO,CAAC,EAAE,OAAO,GACf,OAAO,CAAC,+BAA+B,CAAC,MAAM,CAAC,CAAC,CAgClD"}

package/dist/src/tasks/multimodal/visualQuestionAnswering.d.ts

@@ -1,27 +1,12 @@
+import type { VisualQuestionAnsweringInput, VisualQuestionAnsweringInputData, VisualQuestionAnsweringOutput } from "@huggingface/tasks";
 import type { BaseArgs, Options } from "../../types";
-export type VisualQuestionAnsweringArgs = BaseArgs & {
-    inputs: {
-        /**
-         * Raw image
-         *
-         * You can use native `File` in browsers, or `new Blob([buffer])` in node, or for a base64 image `new Blob([btoa(base64String)])`, or even `await (await fetch('...)).blob()`
-         **/
-        image: Blob | ArrayBuffer;
-        question: string;
+export type VisualQuestionAnsweringArgs = BaseArgs & VisualQuestionAnsweringInput & {
+    inputs: VisualQuestionAnsweringInputData & {
+        image: Blob;
     };
 };
-export interface VisualQuestionAnsweringOutput {
-    /**
-     * A string that’s the answer to a visual question.
-     */
-    answer: string;
-    /**
-     * Answer correctness score.
-     */
-    score: number;
-}
 /**
  * Answers a question on an image. Recommended model: dandelin/vilt-b32-finetuned-vqa.
  */
-export declare function visualQuestionAnswering(args: VisualQuestionAnsweringArgs, options?: Options): Promise<VisualQuestionAnsweringOutput>;
+export declare function visualQuestionAnswering(args: VisualQuestionAnsweringArgs, options?: Options): Promise<VisualQuestionAnsweringOutput[number]>;
 //# sourceMappingURL=visualQuestionAnswering.d.ts.map

package/dist/src/tasks/multimodal/visualQuestionAnswering.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"visualQuestionAnswering.d.ts","sourceRoot":"","sources":["../../../../src/tasks/multimodal/visualQuestionAnswering.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAe,MAAM,aAAa,CAAC;AAIlE,MAAM,MAAM,2BAA2B,GAAG,QAAQ,GAAG;IACpD,MAAM,EAAE;QACP;;;;YAII;QACJ,KAAK,EAAE,IAAI,GAAG,WAAW,CAAC;QAC1B,QAAQ,EAAE,MAAM,CAAC;KACjB,CAAC;CACF,CAAC;AAEF,MAAM,WAAW,6BAA6B;IAC7C;;OAEG;IACH,MAAM,EAAE,MAAM,CAAC;IACf;;OAEG;IACH,KAAK,EAAE,MAAM,CAAC;CACd;AAED;;GAEG;AACH,wBAAsB,uBAAuB,CAC5C,IAAI,EAAE,2BAA2B,EACjC,OAAO,CAAC,EAAE,OAAO,GACf,OAAO,CAAC,6BAA6B,CAAC,CAwBxC"}
+{"version":3,"file":"visualQuestionAnswering.d.ts","sourceRoot":"","sources":["../../../../src/tasks/multimodal/visualQuestionAnswering.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACX,4BAA4B,EAC5B,gCAAgC,EAChC,6BAA6B,EAC7B,MAAM,oBAAoB,CAAC;AAE5B,OAAO,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAe,MAAM,aAAa,CAAC;AAKlE,MAAM,MAAM,2BAA2B,GAAG,QAAQ,GACjD,4BAA4B,GAAG;IAAE,MAAM,EAAE,gCAAgC,GAAG;QAAE,KAAK,EAAE,IAAI,CAAA;KAAE,CAAA;CAAE,CAAC;AAE/F;;GAEG;AACH,wBAAsB,uBAAuB,CAC5C,IAAI,EAAE,2BAA2B,EACjC,OAAO,CAAC,EAAE,OAAO,GACf,OAAO,CAAC,6BAA6B,CAAC,MAAM,CAAC,CAAC,CAsBhD"}

package/dist/src/tasks/nlp/fillMask.d.ts

@@ -1,25 +1,6 @@
+import type { FillMaskInput, FillMaskOutput } from "@huggingface/tasks";
 import type { BaseArgs, Options } from "../../types";
-export type FillMaskArgs = BaseArgs & {
-    inputs: string;
-};
-export type FillMaskOutput = {
-    /**
-     * The probability for this token.
-     */
-    score: number;
-    /**
-     * The actual sequence of tokens that ran against the model (may contain special tokens)
-     */
-    sequence: string;
-    /**
-     * The id of the token
-     */
-    token: number;
-    /**
-     * The string representation of the token
-     */
-    token_str: string;
-}[];
+export type FillMaskArgs = BaseArgs & FillMaskInput;
 /**
  * Tries to fill in a hole with a missing word (token to be precise). That’s the base task for BERT models.
  */

package/dist/src/tasks/nlp/fillMask.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"fillMask.d.ts","sourceRoot":"","sources":["../../../../src/tasks/nlp/fillMask.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,aAAa,CAAC;AAGrD,MAAM,MAAM,YAAY,GAAG,QAAQ,GAAG;IACrC,MAAM,EAAE,MAAM,CAAC;CACf,CAAC;AAEF,MAAM,MAAM,cAAc,GAAG;IAC5B;;OAEG;IACH,KAAK,EAAE,MAAM,CAAC;IACd;;OAEG;IACH,QAAQ,EAAE,MAAM,CAAC;IACjB;;OAEG;IACH,KAAK,EAAE,MAAM,CAAC;IACd;;OAEG;IACH,SAAS,EAAE,MAAM,CAAC;CAClB,EAAE,CAAC;AAEJ;;GAEG;AACH,wBAAsB,QAAQ,CAAC,IAAI,EAAE,YAAY,EAAE,OAAO,CAAC,EAAE,OAAO,GAAG,OAAO,CAAC,cAAc,CAAC,CAoB7F"}
+{"version":3,"file":"fillMask.d.ts","sourceRoot":"","sources":["../../../../src/tasks/nlp/fillMask.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,aAAa,EAAE,cAAc,EAAE,MAAM,oBAAoB,CAAC;AAExE,OAAO,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,aAAa,CAAC;AAGrD,MAAM,MAAM,YAAY,GAAG,QAAQ,GAAG,aAAa,CAAC;AAEpD;;GAEG;AACH,wBAAsB,QAAQ,CAAC,IAAI,EAAE,YAAY,EAAE,OAAO,CAAC,EAAE,OAAO,GAAG,OAAO,CAAC,cAAc,CAAC,CAoB7F"}

package/dist/src/tasks/nlp/questionAnswering.d.ts

@@ -1,30 +1,8 @@
+import type { QuestionAnsweringInput, QuestionAnsweringOutput } from "@huggingface/tasks";
 import type { BaseArgs, Options } from "../../types";
-export type QuestionAnsweringArgs = BaseArgs & {
-    inputs: {
-        context: string;
-        question: string;
-    };
-};
-export interface QuestionAnsweringOutput {
-    /**
-     * A string that’s the answer within the text.
-     */
-    answer: string;
-    /**
-     * The index (string wise) of the stop of the answer within context.
-     */
-    end: number;
-    /**
-     * A float that represents how likely that the answer is correct
-     */
-    score: number;
-    /**
-     * The index (string wise) of the start of the answer within context.
-     */
-    start: number;
-}
+export type QuestionAnsweringArgs = BaseArgs & QuestionAnsweringInput;
 /**
  * Want to have a nice know-it-all bot that can answer any question?. Recommended model: deepset/roberta-base-squad2
  */
-export declare function questionAnswering(args: QuestionAnsweringArgs, options?: Options): Promise<QuestionAnsweringOutput>;
+export declare function questionAnswering(args: QuestionAnsweringArgs, options?: Options): Promise<QuestionAnsweringOutput[number]>;
 //# sourceMappingURL=questionAnswering.d.ts.map

package/dist/src/tasks/nlp/questionAnswering.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"questionAnswering.d.ts","sourceRoot":"","sources":["../../../../src/tasks/nlp/questionAnswering.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,aAAa,CAAC;AAGrD,MAAM,MAAM,qBAAqB,GAAG,QAAQ,GAAG;IAC9C,MAAM,EAAE;QACP,OAAO,EAAE,MAAM,CAAC;QAChB,QAAQ,EAAE,MAAM,CAAC;KACjB,CAAC;CACF,CAAC;AAEF,MAAM,WAAW,uBAAuB;IACvC;;OAEG;IACH,MAAM,EAAE,MAAM,CAAC;IACf;;OAEG;IACH,GAAG,EAAE,MAAM,CAAC;IACZ;;OAEG;IACH,KAAK,EAAE,MAAM,CAAC;IACd;;OAEG;IACH,KAAK,EAAE,MAAM,CAAC;CACd;AAED;;GAEG;AACH,wBAAsB,iBAAiB,CACtC,IAAI,EAAE,qBAAqB,EAC3B,OAAO,CAAC,EAAE,OAAO,GACf,OAAO,CAAC,uBAAuB,CAAC,CAgBlC"}
+{"version":3,"file":"questionAnswering.d.ts","sourceRoot":"","sources":["../../../../src/tasks/nlp/questionAnswering.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,sBAAsB,EAAE,uBAAuB,EAAE,MAAM,oBAAoB,CAAC;AAE1F,OAAO,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,aAAa,CAAC;AAGrD,MAAM,MAAM,qBAAqB,GAAG,QAAQ,GAAG,sBAAsB,CAAC;AAEtE;;GAEG;AACH,wBAAsB,iBAAiB,CACtC,IAAI,EAAE,qBAAqB,EAC3B,OAAO,CAAC,EAAE,OAAO,GACf,OAAO,CAAC,uBAAuB,CAAC,MAAM,CAAC,CAAC,CAyB1C"}

package/dist/src/tasks/nlp/sentenceSimilarity.d.ts

@@ -1,17 +1,6 @@
+import type { SentenceSimilarityInput, SentenceSimilarityOutput } from "@huggingface/tasks";
 import type { BaseArgs, Options } from "../../types";
-export type SentenceSimilarityArgs = BaseArgs & {
-    /**
-     * The inputs vary based on the model.
-     *
-     * For example when using sentence-transformers/paraphrase-xlm-r-multilingual-v1 the inputs will have a `source_sentence` string and
-     * a `sentences` array of strings
-     */
-    inputs: Record<string, unknown> | Record<string, unknown>[];
-};
-/**
- * Returned values are a list of floats
- */
-export type SentenceSimilarityOutput = number[];
+export type SentenceSimilarityArgs = BaseArgs & SentenceSimilarityInput;
 /**
  * Calculate the semantic similarity between one text and a list of other sentences by comparing their embeddings.
  */

package/dist/src/tasks/nlp/sentenceSimilarity.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"sentenceSimilarity.d.ts","sourceRoot":"","sources":["../../../../src/tasks/nlp/sentenceSimilarity.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,aAAa,CAAC;AAGrD,MAAM,MAAM,sBAAsB,GAAG,QAAQ,GAAG;IAC/C;;;;;OAKG;IACH,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAAE,CAAC;CAC5D,CAAC;AAEF;;GAEG;AACH,MAAM,MAAM,wBAAwB,GAAG,MAAM,EAAE,CAAC;AAEhD;;GAEG;AACH,wBAAsB,kBAAkB,CACvC,IAAI,EAAE,sBAAsB,EAC5B,OAAO,CAAC,EAAE,OAAO,GACf,OAAO,CAAC,wBAAwB,CAAC,CAanC"}
+{"version":3,"file":"sentenceSimilarity.d.ts","sourceRoot":"","sources":["../../../../src/tasks/nlp/sentenceSimilarity.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,uBAAuB,EAAE,wBAAwB,EAAE,MAAM,oBAAoB,CAAC;AAG5F,OAAO,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,aAAa,CAAC;AAIrD,MAAM,MAAM,sBAAsB,GAAG,QAAQ,GAAG,uBAAuB,CAAC;AAExE;;GAEG;AACH,wBAAsB,kBAAkB,CACvC,IAAI,EAAE,sBAAsB,EAC5B,OAAO,CAAC,EAAE,OAAO,GACf,OAAO,CAAC,wBAAwB,CAAC,CAanC"}

package/dist/src/tasks/nlp/summarization.d.ts

@@ -1,46 +1,6 @@
+import type { SummarizationInput, SummarizationOutput } from "@huggingface/tasks";
 import type { BaseArgs, Options } from "../../types";
-export type SummarizationArgs = BaseArgs & {
-    /**
-     * A string to be summarized
-     */
-    inputs: string;
-    parameters?: {
-        /**
-         * (Default: None). Integer to define the maximum length in tokens of the output summary.
-         */
-        max_length?: number;
-        /**
-         * (Default: None). Float (0-120.0). The amount of time in seconds that the query should take maximum. Network can cause some overhead so it will be a soft limit.
-         */
-        max_time?: number;
-        /**
-         * (Default: None). Integer to define the minimum length in tokens of the output summary.
-         */
-        min_length?: number;
-        /**
-         * (Default: None). Float (0.0-100.0). The more a token is used within generation the more it is penalized to not be picked in successive generation passes.
-         */
-        repetition_penalty?: number;
-        /**
-         * (Default: 1.0). Float (0.0-100.0). The temperature of the sampling operation. 1 means regular sampling, 0 means always take the highest score, 100.0 is getting closer to uniform probability.
-         */
-        temperature?: number;
-        /**
-         * (Default: None). Integer to define the top tokens considered within the sample operation to create new text.
-         */
-        top_k?: number;
-        /**
-         * (Default: None). Float to define the tokens that are within the sample operation of text generation. Add tokens in the sample for more probable to least probable until the sum of the probabilities is greater than top_p.
-         */
-        top_p?: number;
-    };
-};
-export interface SummarizationOutput {
-    /**
-     * The string after translation
-     */
-    summary_text: string;
-}
+export type SummarizationArgs = BaseArgs & SummarizationInput;
 /**
  * This task is well known to summarize longer text into shorter text. Be careful, some models have a maximum length of input. That means that the summary cannot handle full books for instance. Be careful when choosing your model.
  */

package/dist/src/tasks/nlp/summarization.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"summarization.d.ts","sourceRoot":"","sources":["../../../../src/tasks/nlp/summarization.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,aAAa,CAAC;AAGrD,MAAM,MAAM,iBAAiB,GAAG,QAAQ,GAAG;IAC1C;;OAEG;IACH,MAAM,EAAE,MAAM,CAAC;IACf,UAAU,CAAC,EAAE;QACZ;;WAEG;QACH,UAAU,CAAC,EAAE,MAAM,CAAC;QACpB;;WAEG;QACH,QAAQ,CAAC,EAAE,MAAM,CAAC;QAClB;;WAEG;QACH,UAAU,CAAC,EAAE,MAAM,CAAC;QACpB;;WAEG;QACH,kBAAkB,CAAC,EAAE,MAAM,CAAC;QAC5B;;WAEG;QACH,WAAW,CAAC,EAAE,MAAM,CAAC;QACrB;;WAEG;QACH,KAAK,CAAC,EAAE,MAAM,CAAC;QACf;;WAEG;QACH,KAAK,CAAC,EAAE,MAAM,CAAC;KACf,CAAC;CACF,CAAC;AAEF,MAAM,WAAW,mBAAmB;IACnC;;OAEG;IACH,YAAY,EAAE,MAAM,CAAC;CACrB;AAED;;GAEG;AACH,wBAAsB,aAAa,CAAC,IAAI,EAAE,iBAAiB,EAAE,OAAO,CAAC,EAAE,OAAO,GAAG,OAAO,CAAC,mBAAmB,CAAC,CAU5G"}
+{"version":3,"file":"summarization.d.ts","sourceRoot":"","sources":["../../../../src/tasks/nlp/summarization.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,kBAAkB,EAAE,mBAAmB,EAAE,MAAM,oBAAoB,CAAC;AAElF,OAAO,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,aAAa,CAAC;AAGrD,MAAM,MAAM,iBAAiB,GAAG,QAAQ,GAAG,kBAAkB,CAAC;AAE9D;;GAEG;AACH,wBAAsB,aAAa,CAAC,IAAI,EAAE,iBAAiB,EAAE,OAAO,CAAC,EAAE,OAAO,GAAG,OAAO,CAAC,mBAAmB,CAAC,CAU5G"}

package/dist/src/tasks/nlp/tableQuestionAnswering.d.ts

@@ -1,36 +1,8 @@
+import type { TableQuestionAnsweringInput, TableQuestionAnsweringOutput } from "@huggingface/tasks";
 import type { BaseArgs, Options } from "../../types";
-export type TableQuestionAnsweringArgs = BaseArgs & {
-    inputs: {
-        /**
-         * The query in plain text that you want to ask the table
-         */
-        query: string;
-        /**
-         * A table of data represented as a dict of list where entries are headers and the lists are all the values, all lists must have the same size.
-         */
-        table: Record<string, string[]>;
-    };
-};
-export interface TableQuestionAnsweringOutput {
-    /**
-     * The aggregator used to get the answer
-     */
-    aggregator: string;
-    /**
-     * The plaintext answer
-     */
-    answer: string;
-    /**
-     * A list of coordinates of the cells contents
-     */
-    cells: string[];
-    /**
-     * a list of coordinates of the cells referenced in the answer
-     */
-    coordinates: number[][];
-}
+export type TableQuestionAnsweringArgs = BaseArgs & TableQuestionAnsweringInput;
 /**
  * Don’t know SQL? Don’t want to dive into a large spreadsheet? Ask questions in plain english! Recommended model: google/tapas-base-finetuned-wtq.
  */
-export declare function tableQuestionAnswering(args: TableQuestionAnsweringArgs, options?: Options): Promise<TableQuestionAnsweringOutput>;
+export declare function tableQuestionAnswering(args: TableQuestionAnsweringArgs, options?: Options): Promise<TableQuestionAnsweringOutput[number]>;
 //# sourceMappingURL=tableQuestionAnswering.d.ts.map

package/dist/src/tasks/nlp/tableQuestionAnswering.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"tableQuestionAnswering.d.ts","sourceRoot":"","sources":["../../../../src/tasks/nlp/tableQuestionAnswering.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,aAAa,CAAC;AAGrD,MAAM,MAAM,0BAA0B,GAAG,QAAQ,GAAG;IACnD,MAAM,EAAE;QACP;;WAEG;QACH,KAAK,EAAE,MAAM,CAAC;QACd;;WAEG;QACH,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,EAAE,CAAC,CAAC;KAChC,CAAC;CACF,CAAC;AAEF,MAAM,WAAW,4BAA4B;IAC5C;;OAEG;IACH,UAAU,EAAE,MAAM,CAAC;IACnB;;OAEG;IACH,MAAM,EAAE,MAAM,CAAC;IACf;;OAEG;IACH,KAAK,EAAE,MAAM,EAAE,CAAC;IAChB;;OAEG;IACH,WAAW,EAAE,MAAM,EAAE,EAAE,CAAC;CACxB;AAED;;GAEG;AACH,wBAAsB,sBAAsB,CAC3C,IAAI,EAAE,0BAA0B,EAChC,OAAO,CAAC,EAAE,OAAO,GACf,OAAO,CAAC,4BAA4B,CAAC,CAkBvC"}
+{"version":3,"file":"tableQuestionAnswering.d.ts","sourceRoot":"","sources":["../../../../src/tasks/nlp/tableQuestionAnswering.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,2BAA2B,EAAE,4BAA4B,EAAE,MAAM,oBAAoB,CAAC;AAEpG,OAAO,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,aAAa,CAAC;AAGrD,MAAM,MAAM,0BAA0B,GAAG,QAAQ,GAAG,2BAA2B,CAAC;AAEhF;;GAEG;AACH,wBAAsB,sBAAsB,CAC3C,IAAI,EAAE,0BAA0B,EAChC,OAAO,CAAC,EAAE,OAAO,GACf,OAAO,CAAC,4BAA4B,CAAC,MAAM,CAAC,CAAC,CAY/C"}

package/dist/src/tasks/nlp/textClassification.d.ts

@@ -1,20 +1,6 @@
+import type { TextClassificationInput, TextClassificationOutput } from "@huggingface/tasks";
 import type { BaseArgs, Options } from "../../types";
-export type TextClassificationArgs = BaseArgs & {
-    /**
-     * A string to be classified
-     */
-    inputs: string;
-};
-export type TextClassificationOutput = {
-    /**
-     * The label for the class (model specific)
-     */
-    label: string;
-    /**
-     * A floats that represents how likely is that the text belongs to this class.
-     */
-    score: number;
-}[];
+export type TextClassificationArgs = BaseArgs & TextClassificationInput;
 /**
  * Usually used for sentiment-analysis this will output the likelihood of classes of an input. Recommended model: distilbert-base-uncased-finetuned-sst-2-english
  */

package/dist/src/tasks/nlp/textClassification.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"textClassification.d.ts","sourceRoot":"","sources":["../../../../src/tasks/nlp/textClassification.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,aAAa,CAAC;AAGrD,MAAM,MAAM,sBAAsB,GAAG,QAAQ,GAAG;IAC/C;;OAEG;IACH,MAAM,EAAE,MAAM,CAAC;CACf,CAAC;AAEF,MAAM,MAAM,wBAAwB,GAAG;IACtC;;OAEG;IACH,KAAK,EAAE,MAAM,CAAC;IACd;;OAEG;IACH,KAAK,EAAE,MAAM,CAAC;CACd,EAAE,CAAC;AAEJ;;GAEG;AACH,wBAAsB,kBAAkB,CACvC,IAAI,EAAE,sBAAsB,EAC5B,OAAO,CAAC,EAAE,OAAO,GACf,OAAO,CAAC,wBAAwB,CAAC,CAanC"}
+{"version":3,"file":"textClassification.d.ts","sourceRoot":"","sources":["../../../../src/tasks/nlp/textClassification.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,uBAAuB,EAAE,wBAAwB,EAAE,MAAM,oBAAoB,CAAC;AAE5F,OAAO,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,aAAa,CAAC;AAGrD,MAAM,MAAM,sBAAsB,GAAG,QAAQ,GAAG,uBAAuB,CAAC;AAExE;;GAEG;AACH,wBAAsB,kBAAkB,CACvC,IAAI,EAAE,sBAAsB,EAC5B,OAAO,CAAC,EAAE,OAAO,GACf,OAAO,CAAC,wBAAwB,CAAC,CAanC"}