@huggingface/tasks 0.13.1 → 0.13.2

package/dist/commonjs/tasks/image-feature-extraction/data.js

@@ -0,0 +1,57 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const taskData = {
+    datasets: [
+        {
+            description: "ImageNet-1K is a image classification dataset in which images are used to train image-feature-extraction models.",
+            id: "imagenet-1k",
+        },
+    ],
+    demo: {
+        inputs: [
+            {
+                filename: "mask-generation-input.png",
+                type: "img",
+            },
+        ],
+        outputs: [
+            {
+                table: [
+                    ["Dimension 1", "Dimension 2", "Dimension 3"],
+                    ["0.21236686408519745", "1.0919708013534546", "0.8512550592422485"],
+                    ["0.809657871723175", "-0.18544459342956543", "-0.7851548194885254"],
+                    ["1.3103108406066895", "-0.2479034662246704", "-0.9107287526130676"],
+                    ["1.8536205291748047", "-0.36419737339019775", "0.09717650711536407"],
+                ],
+                type: "tabular",
+            },
+        ],
+    },
+    metrics: [],
+    models: [
+        {
+            description: "A powerful image feature extraction model.",
+            id: "timm/vit_large_patch14_dinov2.lvd142m",
+        },
+        {
+            description: "A strong image feature extraction model.",
+            id: "nvidia/MambaVision-T-1K",
+        },
+        {
+            description: "A robust image feature extraction model.",
+            id: "facebook/dino-vitb16",
+        },
+        {
+            description: "Strong image feature extraction model made for information retrieval from documents.",
+            id: "vidore/colpali",
+        },
+        {
+            description: "Strong image feature extraction model that can be used on images and documents.",
+            id: "OpenGVLab/InternViT-6B-448px-V1-2",
+        },
+    ],
+    spaces: [],
+    summary: "Image feature extraction is the task of extracting features learnt in a computer vision model.",
+    widgetModels: [],
+};
+exports.default = taskData;

package/dist/commonjs/tasks/image-segmentation/data.d.ts

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

package/dist/{src → commonjs}/tasks/image-segmentation/data.d.ts.map

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

package/dist/commonjs/tasks/image-segmentation/data.js

@@ -0,0 +1,95 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const taskData = {
+    datasets: [
+        {
+            description: "Scene segmentation dataset.",
+            id: "scene_parse_150",
+        },
+    ],
+    demo: {
+        inputs: [
+            {
+                filename: "image-segmentation-input.jpeg",
+                type: "img",
+            },
+        ],
+        outputs: [
+            {
+                filename: "image-segmentation-output.png",
+                type: "img",
+            },
+        ],
+    },
+    metrics: [
+        {
+            description: "Average Precision (AP) is the Area Under the PR Curve (AUC-PR). It is calculated for each semantic class separately",
+            id: "Average Precision",
+        },
+        {
+            description: "Mean Average Precision (mAP) is the overall average of the AP values",
+            id: "Mean Average Precision",
+        },
+        {
+            description: "Intersection over Union (IoU) is the overlap of segmentation masks. Mean IoU is the average of the IoU of all semantic classes",
+            id: "Mean Intersection over Union",
+        },
+        {
+            description: "APα is the Average Precision at the IoU threshold of a α value, for example, AP50 and AP75",
+            id: "APα",
+        },
+    ],
+    models: [
+        {
+            // TO DO: write description
+            description: "Solid semantic segmentation model trained on ADE20k.",
+            id: "openmmlab/upernet-convnext-small",
+        },
+        {
+            description: "Background removal model.",
+            id: "briaai/RMBG-1.4",
+        },
+        {
+            description: "A multipurpose image segmentation model for high resolution images.",
+            id: "ZhengPeng7/BiRefNet",
+        },
+        {
+            description: "Powerful human-centric image segmentation model.",
+            id: "facebook/sapiens-seg-1b",
+        },
+        {
+            description: "Panoptic segmentation model trained on the COCO (common objects) dataset.",
+            id: "facebook/mask2former-swin-large-coco-panoptic",
+        },
+    ],
+    spaces: [
+        {
+            description: "A semantic segmentation application that can predict unseen instances out of the box.",
+            id: "facebook/ov-seg",
+        },
+        {
+            description: "One of the strongest segmentation applications.",
+            id: "jbrinkma/segment-anything",
+        },
+        {
+            description: "A human-centric segmentation model.",
+            id: "facebook/sapiens-pose",
+        },
+        {
+            description: "An instance segmentation application to predict neuronal cell types from microscopy images.",
+            id: "rashmi/sartorius-cell-instance-segmentation",
+        },
+        {
+            description: "An application that segments videos.",
+            id: "ArtGAN/Segment-Anything-Video",
+        },
+        {
+            description: "An panoptic segmentation application built for outdoor environments.",
+            id: "segments/panoptic-segment-anything",
+        },
+    ],
+    summary: "Image Segmentation divides an image into segments where each pixel in the image is mapped to an object. This task has multiple variants such as instance segmentation, panoptic segmentation and semantic segmentation.",
+    widgetModels: ["nvidia/segformer-b0-finetuned-ade-512-512"],
+    youtubeId: "dKE8SIt9C-w",
+};
+exports.default = taskData;

package/dist/commonjs/tasks/image-segmentation/inference.js

@@ -0,0 +1,2 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });

package/dist/commonjs/tasks/image-text-to-text/data.d.ts

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

package/dist/{src → commonjs}/tasks/image-text-to-text/data.d.ts.map

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

package/dist/commonjs/tasks/image-text-to-text/data.js

@@ -0,0 +1,99 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const taskData = {
+    datasets: [
+        {
+            description: "Instructions composed of image and text.",
+            id: "liuhaotian/LLaVA-Instruct-150K",
+        },
+        {
+            description: "Conversation turns where questions involve image and text.",
+            id: "liuhaotian/LLaVA-Pretrain",
+        },
+        {
+            description: "A collection of datasets made for model fine-tuning.",
+            id: "HuggingFaceM4/the_cauldron",
+        },
+        {
+            description: "Screenshots of websites with their HTML/CSS codes.",
+            id: "HuggingFaceM4/WebSight",
+        },
+    ],
+    demo: {
+        inputs: [
+            {
+                filename: "image-text-to-text-input.png",
+                type: "img",
+            },
+            {
+                label: "Text Prompt",
+                content: "Describe the position of the bee in detail.",
+                type: "text",
+            },
+        ],
+        outputs: [
+            {
+                label: "Answer",
+                content: "The bee is sitting on a pink flower, surrounded by other flowers. The bee is positioned in the center of the flower, with its head and front legs sticking out.",
+                type: "text",
+            },
+        ],
+    },
+    metrics: [],
+    models: [
+        {
+            description: "Powerful vision language model with great visual understanding and reasoning capabilities.",
+            id: "meta-llama/Llama-3.2-11B-Vision-Instruct",
+        },
+        {
+            description: "Cutting-edge vision language models.",
+            id: "allenai/Molmo-7B-D-0924",
+        },
+        {
+            description: "Small yet powerful model.",
+            id: "vikhyatk/moondream2",
+        },
+        {
+            description: "Strong image-text-to-text model.",
+            id: "Qwen/Qwen2-VL-7B-Instruct",
+        },
+        {
+            description: "Strong image-text-to-text model.",
+            id: "mistralai/Pixtral-12B-2409",
+        },
+        {
+            description: "Strong image-text-to-text model focused on documents.",
+            id: "stepfun-ai/GOT-OCR2_0",
+        },
+    ],
+    spaces: [
+        {
+            description: "Leaderboard to evaluate vision language models.",
+            id: "opencompass/open_vlm_leaderboard",
+        },
+        {
+            description: "Vision language models arena, where models are ranked by votes of users.",
+            id: "WildVision/vision-arena",
+        },
+        {
+            description: "Powerful vision-language model assistant.",
+            id: "akhaliq/Molmo-7B-D-0924",
+        },
+        {
+            description: "An image-text-to-text application focused on documents.",
+            id: "stepfun-ai/GOT_official_online_demo",
+        },
+        {
+            description: "An application to compare outputs of different vision language models.",
+            id: "merve/compare_VLMs",
+        },
+        {
+            description: "An application for chatting with an image-text-to-text model.",
+            id: "GanymedeNil/Qwen2-VL-7B",
+        },
+    ],
+    summary: "Image-text-to-text models take in an image and text prompt and output text. These models are also called vision-language models, or VLMs. The difference from image-to-text models is that these models take an additional text input, not restricting the model to certain use cases like image captioning, and may also be trained to accept a conversation as input.",
+    widgetModels: ["meta-llama/Llama-3.2-11B-Vision-Instruct"],
+    youtubeId: "IoGaGfU1CIg",
+};
+exports.default = taskData;

package/dist/commonjs/tasks/image-to-3d/data.d.ts

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

package/dist/{src → commonjs}/tasks/image-to-3d/data.d.ts.map

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

package/dist/commonjs/tasks/image-to-3d/data.js

@@ -0,0 +1,74 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const taskData = {
+    datasets: [
+        {
+            description: "A large dataset of over 10 million 3D objects.",
+            id: "allenai/objaverse-xl",
+        },
+        {
+            description: "A dataset of isolated object images for evaluating image-to-3D models.",
+            id: "dylanebert/iso3d",
+        },
+    ],
+    demo: {
+        inputs: [
+            {
+                filename: "image-to-3d-image-input.png",
+                type: "img",
+            },
+        ],
+        outputs: [
+            {
+                label: "Result",
+                content: "image-to-3d-3d-output-filename.glb",
+                type: "text",
+            },
+        ],
+    },
+    metrics: [],
+    models: [
+        {
+            description: "Fast image-to-3D mesh model by Tencent.",
+            id: "TencentARC/InstantMesh",
+        },
+        {
+            description: "Fast image-to-3D mesh model by StabilityAI",
+            id: "stabilityai/TripoSR",
+        },
+        {
+            description: "A scaled up image-to-3D mesh model derived from TripoSR.",
+            id: "hwjiang/Real3D",
+        },
+        {
+            description: "Generative 3D gaussian splatting model.",
+            id: "ashawkey/LGM",
+        },
+    ],
+    spaces: [
+        {
+            description: "Leaderboard to evaluate image-to-3D models.",
+            id: "dylanebert/3d-arena",
+        },
+        {
+            description: "Image-to-3D demo with mesh outputs.",
+            id: "TencentARC/InstantMesh",
+        },
+        {
+            description: "Image-to-3D demo with mesh outputs.",
+            id: "stabilityai/TripoSR",
+        },
+        {
+            description: "Image-to-3D demo with mesh outputs.",
+            id: "hwjiang/Real3D",
+        },
+        {
+            description: "Image-to-3D demo with splat outputs.",
+            id: "dylanebert/LGM-mini",
+        },
+    ],
+    summary: "Image-to-3D models take in image input and produce 3D output.",
+    widgetModels: [],
+    youtubeId: "",
+};
+exports.default = taskData;

package/dist/commonjs/tasks/image-to-image/data.d.ts

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

package/dist/{src → commonjs}/tasks/image-to-image/data.d.ts.map

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

package/dist/commonjs/tasks/image-to-image/data.js

@@ -0,0 +1,95 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const taskData = {
+    datasets: [
+        {
+            description: "Synthetic dataset, for image relighting",
+            id: "VIDIT",
+        },
+        {
+            description: "Multiple images of celebrities, used for facial expression translation",
+            id: "huggan/CelebA-faces",
+        },
+    ],
+    demo: {
+        inputs: [
+            {
+                filename: "image-to-image-input.jpeg",
+                type: "img",
+            },
+        ],
+        outputs: [
+            {
+                filename: "image-to-image-output.png",
+                type: "img",
+            },
+        ],
+    },
+    isPlaceholder: false,
+    metrics: [
+        {
+            description: "Peak Signal to Noise Ratio (PSNR) is an approximation of the human perception, considering the ratio of the absolute intensity with respect to the variations. Measured in dB, a high value indicates a high fidelity.",
+            id: "PSNR",
+        },
+        {
+            description: "Structural Similarity Index (SSIM) is a perceptual metric which compares the luminance, contrast and structure of two images. The values of SSIM range between -1 and 1, and higher values indicate closer resemblance to the original image.",
+            id: "SSIM",
+        },
+        {
+            description: "Inception Score (IS) is an analysis of the labels predicted by an image classification model when presented with a sample of the generated images.",
+            id: "IS",
+        },
+    ],
+    models: [
+        {
+            description: "An image-to-image model to improve image resolution.",
+            id: "fal/AuraSR-v2",
+        },
+        {
+            description: "A model that increases the resolution of an image.",
+            id: "keras-io/super-resolution",
+        },
+        {
+            description: "A model that creates a set of variations of the input image in the style of DALL-E using Stable Diffusion.",
+            id: "lambdalabs/sd-image-variations-diffusers",
+        },
+        {
+            description: "A model that generates images based on segments in the input image and the text prompt.",
+            id: "mfidabel/controlnet-segment-anything",
+        },
+        {
+            description: "A model that takes an image and an instruction to edit the image.",
+            id: "timbrooks/instruct-pix2pix",
+        },
+    ],
+    spaces: [
+        {
+            description: "Image enhancer application for low light.",
+            id: "keras-io/low-light-image-enhancement",
+        },
+        {
+            description: "Style transfer application.",
+            id: "keras-io/neural-style-transfer",
+        },
+        {
+            description: "An application that generates images based on segment control.",
+            id: "mfidabel/controlnet-segment-anything",
+        },
+        {
+            description: "Image generation application that takes image control and text prompt.",
+            id: "hysts/ControlNet",
+        },
+        {
+            description: "Colorize any image using this app.",
+            id: "ioclab/brightness-controlnet",
+        },
+        {
+            description: "Edit images with instructions.",
+            id: "timbrooks/instruct-pix2pix",
+        },
+    ],
+    summary: "Image-to-image is the task of transforming an input image through a variety of possible manipulations and enhancements, such as super-resolution, image inpainting, colorization, and more.",
+    widgetModels: ["stabilityai/stable-diffusion-2-inpainting"],
+    youtubeId: "",
+};
+exports.default = taskData;

package/dist/commonjs/tasks/image-to-image/inference.js

@@ -0,0 +1,7 @@
+"use strict";
+/**
+ * Inference code generated from the JSON schema spec in ./spec
+ *
+ * Using src/scripts/inference-codegen
+ */
+Object.defineProperty(exports, "__esModule", { value: true });

package/dist/commonjs/tasks/image-to-text/data.d.ts

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

package/dist/{src → commonjs}/tasks/image-to-text/data.d.ts.map

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

package/dist/commonjs/tasks/image-to-text/data.js

@@ -0,0 +1,80 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const taskData = {
+    datasets: [
+        {
+            // TODO write proper description
+            description: "Dataset from 12M image-text of Reddit",
+            id: "red_caps",
+        },
+        {
+            // TODO write proper description
+            description: "Dataset from 3.3M images of Google",
+            id: "datasets/conceptual_captions",
+        },
+    ],
+    demo: {
+        inputs: [
+            {
+                filename: "savanna.jpg",
+                type: "img",
+            },
+        ],
+        outputs: [
+            {
+                label: "Detailed description",
+                content: "a herd of giraffes and zebras grazing in a field",
+                type: "text",
+            },
+        ],
+    },
+    metrics: [],
+    models: [
+        {
+            description: "A robust image captioning model.",
+            id: "Salesforce/blip2-opt-2.7b",
+        },
+        {
+            description: "A powerful and accurate image-to-text model that can also localize concepts in images.",
+            id: "microsoft/kosmos-2-patch14-224",
+        },
+        {
+            description: "A strong optical character recognition model.",
+            id: "facebook/nougat-base",
+        },
+        {
+            description: "A powerful model that lets you have a conversation with the image.",
+            id: "llava-hf/llava-1.5-7b-hf",
+        },
+    ],
+    spaces: [
+        {
+            description: "An application that compares various image captioning models.",
+            id: "nielsr/comparing-captioning-models",
+        },
+        {
+            description: "A robust image captioning application.",
+            id: "flax-community/image-captioning",
+        },
+        {
+            description: "An application that transcribes handwritings into text.",
+            id: "nielsr/TrOCR-handwritten",
+        },
+        {
+            description: "An application that can caption images and answer questions about a given image.",
+            id: "Salesforce/BLIP",
+        },
+        {
+            description: "An application that can caption images and answer questions with a conversational agent.",
+            id: "Salesforce/BLIP2",
+        },
+        {
+            description: "An image captioning application that demonstrates the effect of noise on captions.",
+            id: "johko/capdec-image-captioning",
+        },
+    ],
+    summary: "Image to text models output a text from a given image. Image captioning or optical character recognition can be considered as the most common applications of image to text.",
+    widgetModels: ["Salesforce/blip-image-captioning-large"],
+    youtubeId: "",
+};
+exports.default = taskData;

package/dist/commonjs/tasks/image-to-text/inference.js

@@ -0,0 +1,7 @@
+"use strict";
+/**
+ * Inference code generated from the JSON schema spec in ./spec
+ *
+ * Using src/scripts/inference-codegen
+ */
+Object.defineProperty(exports, "__esModule", { value: true });

package/dist/{src → commonjs}/tasks/index.d.ts

@@ -1,32 +1,32 @@
-import type { PipelineType } from "../pipelines";
-export type * from "./audio-classification/inference";
-export type * from "./automatic-speech-recognition/inference";
-export type { ChatCompletionInput, ChatCompletionInputMessage, ChatCompletionOutput, ChatCompletionOutputComplete, ChatCompletionOutputMessage, ChatCompletionStreamOutput, ChatCompletionStreamOutputChoice, ChatCompletionStreamOutputDelta, } from "./chat-completion/inference";
-export type * from "./document-question-answering/inference";
-export type * from "./feature-extraction/inference";
-export type * from "./fill-mask/inference";
-export type { ImageClassificationInput, ImageClassificationOutput, ImageClassificationOutputElement, ImageClassificationParameters, } from "./image-classification/inference";
-export type * from "./image-to-image/inference";
-export type { ImageToTextInput, ImageToTextOutput, ImageToTextParameters } from "./image-to-text/inference";
-export type * from "./image-segmentation/inference";
-export type * from "./object-detection/inference";
-export type * from "./depth-estimation/inference";
-export type * from "./question-answering/inference";
-export type * from "./sentence-similarity/inference";
-export type * from "./summarization/inference";
-export type * from "./table-question-answering/inference";
-export type { TextToImageInput, TextToImageOutput, TextToImageParameters } from "./text-to-image/inference";
-export type { TextToSpeechParameters, TextToSpeechInput, TextToSpeechOutput } from "./text-to-speech/inference";
-export type * from "./token-classification/inference";
-export type { TranslationInput, TranslationOutput } from "./translation/inference";
-export type { ClassificationOutputTransform, TextClassificationInput, TextClassificationOutput, TextClassificationOutputElement, TextClassificationParameters, } from "./text-classification/inference";
-export type { TextGenerationOutputFinishReason, TextGenerationOutputPrefillToken, TextGenerationInput, TextGenerationOutput, TextGenerationOutputDetails, TextGenerationInputGenerateParameters, TextGenerationOutputBestOfSequence, TextGenerationOutputToken, TextGenerationStreamOutputStreamDetails, TextGenerationStreamOutput, } from "./text-generation/inference";
-export type * from "./video-classification/inference";
-export type * from "./visual-question-answering/inference";
-export type * from "./zero-shot-classification/inference";
-export type * from "./zero-shot-image-classification/inference";
-export type { BoundingBox, ZeroShotObjectDetectionInput, ZeroShotObjectDetectionInputData, ZeroShotObjectDetectionOutput, ZeroShotObjectDetectionOutputElement, } from "./zero-shot-object-detection/inference";
-import type { ModelLibraryKey } from "../model-libraries";
+import type { PipelineType } from "../pipelines.js";
+export type * from "./audio-classification/inference.js";
+export type * from "./automatic-speech-recognition/inference.js";
+export type { ChatCompletionInput, ChatCompletionInputMessage, ChatCompletionOutput, ChatCompletionOutputComplete, ChatCompletionOutputMessage, ChatCompletionStreamOutput, ChatCompletionStreamOutputChoice, ChatCompletionStreamOutputDelta, } from "./chat-completion/inference.js";
+export type * from "./document-question-answering/inference.js";
+export type * from "./feature-extraction/inference.js";
+export type * from "./fill-mask/inference.js";
+export type { ImageClassificationInput, ImageClassificationOutput, ImageClassificationOutputElement, ImageClassificationParameters, } from "./image-classification/inference.js";
+export type * from "./image-to-image/inference.js";
+export type { ImageToTextInput, ImageToTextOutput, ImageToTextParameters } from "./image-to-text/inference.js";
+export type * from "./image-segmentation/inference.js";
+export type * from "./object-detection/inference.js";
+export type * from "./depth-estimation/inference.js";
+export type * from "./question-answering/inference.js";
+export type * from "./sentence-similarity/inference.js";
+export type * from "./summarization/inference.js";
+export type * from "./table-question-answering/inference.js";
+export type { TextToImageInput, TextToImageOutput, TextToImageParameters } from "./text-to-image/inference.js";
+export type { TextToSpeechParameters, TextToSpeechInput, TextToSpeechOutput } from "./text-to-speech/inference.js";
+export type * from "./token-classification/inference.js";
+export type { TranslationInput, TranslationOutput } from "./translation/inference.js";
+export type { ClassificationOutputTransform, TextClassificationInput, TextClassificationOutput, TextClassificationOutputElement, TextClassificationParameters, } from "./text-classification/inference.js";
+export type { TextGenerationOutputFinishReason, TextGenerationOutputPrefillToken, TextGenerationInput, TextGenerationOutput, TextGenerationOutputDetails, TextGenerationInputGenerateParameters, TextGenerationOutputBestOfSequence, TextGenerationOutputToken, TextGenerationStreamOutputStreamDetails, TextGenerationStreamOutput, } from "./text-generation/inference.js";
+export type * from "./video-classification/inference.js";
+export type * from "./visual-question-answering/inference.js";
+export type * from "./zero-shot-classification/inference.js";
+export type * from "./zero-shot-image-classification/inference.js";
+export type { BoundingBox, ZeroShotObjectDetectionInput, ZeroShotObjectDetectionInputData, ZeroShotObjectDetectionOutput, ZeroShotObjectDetectionOutputElement, } from "./zero-shot-object-detection/inference.js";
+import type { ModelLibraryKey } from "../model-libraries.js";
 /**
  * Model libraries compatible with each ML task
  */

package/dist/commonjs/tasks/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/tasks/index.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,iBAAiB,CAAC;AA4CpD,mBAAmB,qCAAqC,CAAC;AACzD,mBAAmB,6CAA6C,CAAC;AACjE,YAAY,EACX,mBAAmB,EACnB,0BAA0B,EAC1B,oBAAoB,EACpB,4BAA4B,EAC5B,2BAA2B,EAC3B,0BAA0B,EAC1B,gCAAgC,EAChC,+BAA+B,GAC/B,MAAM,gCAAgC,CAAC;AACxC,mBAAmB,4CAA4C,CAAC;AAChE,mBAAmB,mCAAmC,CAAC;AACvD,mBAAmB,0BAA0B,CAAC;AAC9C,YAAY,EACX,wBAAwB,EACxB,yBAAyB,EACzB,gCAAgC,EAChC,6BAA6B,GAC7B,MAAM,qCAAqC,CAAC;AAC7C,mBAAmB,+BAA+B,CAAC;AACnD,YAAY,EAAE,gBAAgB,EAAE,iBAAiB,EAAE,qBAAqB,EAAE,MAAM,8BAA8B,CAAC;AAC/G,mBAAmB,mCAAmC,CAAC;AACvD,mBAAmB,iCAAiC,CAAC;AACrD,mBAAmB,iCAAiC,CAAC;AACrD,mBAAmB,mCAAmC,CAAC;AACvD,mBAAmB,oCAAoC,CAAC;AACxD,mBAAmB,8BAA8B,CAAC;AAClD,mBAAmB,yCAAyC,CAAC;AAC7D,YAAY,EAAE,gBAAgB,EAAE,iBAAiB,EAAE,qBAAqB,EAAE,MAAM,8BAA8B,CAAC;AAC/G,YAAY,EAAE,sBAAsB,EAAE,iBAAiB,EAAE,kBAAkB,EAAE,MAAM,+BAA+B,CAAC;AACnH,mBAAmB,qCAAqC,CAAC;AACzD,YAAY,EAAE,gBAAgB,EAAE,iBAAiB,EAAE,MAAM,4BAA4B,CAAC;AACtF,YAAY,EACX,6BAA6B,EAC7B,uBAAuB,EACvB,wBAAwB,EACxB,+BAA+B,EAC/B,4BAA4B,GAC5B,MAAM,oCAAoC,CAAC;AAC5C,YAAY,EACX,gCAAgC,EAChC,gCAAgC,EAChC,mBAAmB,EACnB,oBAAoB,EACpB,2BAA2B,EAC3B,qCAAqC,EACrC,kCAAkC,EAClC,yBAAyB,EACzB,uCAAuC,EACvC,0BAA0B,GAC1B,MAAM,gCAAgC,CAAC;AACxC,mBAAmB,qCAAqC,CAAC;AACzD,mBAAmB,0CAA0C,CAAC;AAC9D,mBAAmB,yCAAyC,CAAC;AAC7D,mBAAmB,+CAA+C,CAAC;AACnE,YAAY,EACX,WAAW,EACX,4BAA4B,EAC5B,gCAAgC,EAChC,6BAA6B,EAC7B,oCAAoC,GACpC,MAAM,2CAA2C,CAAC;AAEnD,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,uBAAuB,CAAC;AAE7D;;GAEG;AACH,eAAO,MAAM,qBAAqB,EAAE,MAAM,CAAC,YAAY,EAAE,eAAe,EAAE,CA6DzE,CAAC;AAoBF,eAAO,MAAM,UAAU,EAAE,MAAM,CAAC,YAAY,EAAE,QAAQ,GAAG,SAAS,CAqDxD,CAAC;AAEX,MAAM,WAAW,WAAW;IAC3B,WAAW,EAAE,MAAM,CAAC;IACpB,EAAE,EAAE,MAAM,CAAC;CACX;AAED,MAAM,MAAM,aAAa,GACtB;IACA,QAAQ,EAAE,MAAM,CAAC;IACjB,IAAI,EAAE,OAAO,CAAC;CACb,GACD;IACA,IAAI,EAAE,KAAK,CAAC;QACX,KAAK,EAAE,MAAM,CAAC;QACd,KAAK,EAAE,MAAM,CAAC;KACd,CAAC,CAAC;IACH,IAAI,EAAE,OAAO,CAAC;CACb,GACD;IACA,QAAQ,EAAE,MAAM,CAAC;IACjB,IAAI,EAAE,KAAK,CAAC;CACX,GACD;IACA,KAAK,EAAE,MAAM,EAAE,EAAE,CAAC;IAClB,IAAI,EAAE,SAAS,CAAC;CACf,GACD;IACA,OAAO,EAAE,MAAM,CAAC;IAChB,KAAK,EAAE,MAAM,CAAC;IACd,IAAI,EAAE,MAAM,CAAC;CACZ,GACD;IACA,IAAI,EAAE,MAAM,CAAC;IACb,MAAM,EAAE,KAAK,CAAC;QACb,GAAG,EAAE,MAAM,CAAC;QACZ,KAAK,EAAE,MAAM,CAAC;QACd,IAAI,EAAE,MAAM,CAAC;KACb,CAAC,CAAC;IACH,IAAI,EAAE,kBAAkB,CAAC;CACxB,CAAC;AAEL,MAAM,WAAW,QAAQ;IACxB,MAAM,EAAE,aAAa,EAAE,CAAC;IACxB,OAAO,EAAE,aAAa,EAAE,CAAC;CACzB;AAED,MAAM,WAAW,QAAQ;IACxB,QAAQ,EAAE,WAAW,EAAE,CAAC;IACxB,IAAI,EAAE,QAAQ,CAAC;IACf,EAAE,EAAE,YAAY,CAAC;IACjB,WAAW,CAAC,EAAE,YAAY,CAAC;IAC3B,aAAa,CAAC,EAAE,OAAO,CAAC;IACxB,KAAK,EAAE,MAAM,CAAC;IACd,SAAS,EAAE,eAAe,EAAE,CAAC;IAC7B,OAAO,EAAE,WAAW,EAAE,CAAC;IACvB,MAAM,EAAE,WAAW,EAAE,CAAC;IACtB,MAAM,EAAE,WAAW,EAAE,CAAC;IACtB,OAAO,EAAE,MAAM,CAAC;IAChB,YAAY,EAAE,MAAM,EAAE,CAAC;IACvB,SAAS,CAAC,EAAE,MAAM,CAAC;CACnB;AAED,MAAM,MAAM,cAAc,GAAG,IAAI,CAAC,QAAQ,EAAE,IAAI,GAAG,OAAO,GAAG,WAAW,CAAC,CAAC"}