@huggingface/tasks 0.0.9 → 0.1.1

package/dist/index.d.ts

@@ -23,6 +23,7 @@ declare enum ModelLibrary {
     "pyannote-audio" = "pyannote.audio",
     "sample-factory" = "Sample Factory",
     "sentence-transformers" = "Sentence Transformers",
+    "setfit" = "SetFit",
     "sklearn" = "Scikit-learn",
     "spacy" = "spaCy",
     "span-marker" = "SpanMarker",
@@ -452,6 +453,9 @@ declare const PIPELINE_TYPES_SET: Set<"other" | "text-classification" | "token-c
  */
 declare const LIBRARY_TASK_MAPPING_EXCLUDING_TRANSFORMERS: Partial<Record<ModelLibraryKey, PipelineType[]>>;
 
+/**
+ * See default-widget-inputs.ts for the default widget inputs, this files only contains the types
+ */
 type TableData = Record<string, (string | number)[]>;
 type WidgetExampleOutputLabels = Array<{
     label: string;
@@ -526,6 +530,7 @@ interface WidgetExampleSentenceSimilarityInput<TOutput = WidgetExampleOutput> ex
 type WidgetExample<TOutput = WidgetExampleOutput> = WidgetExampleTextInput<TOutput> | WidgetExampleTextAndContextInput<TOutput> | WidgetExampleTextAndTableInput<TOutput> | WidgetExampleAssetInput<TOutput> | WidgetExampleAssetAndPromptInput<TOutput> | WidgetExampleAssetAndTextInput<TOutput> | WidgetExampleAssetAndZeroShotInput<TOutput> | WidgetExampleStructuredDataInput<TOutput> | WidgetExampleTableDataInput<TOutput> | WidgetExampleZeroShotTextInput<TOutput> | WidgetExampleSentenceSimilarityInput<TOutput>;
 type KeysOfUnion<T> = T extends unknown ? keyof T : never;
 type WidgetExampleAttribute = KeysOfUnion<WidgetExample>;
+
 declare enum InferenceDisplayability {
     /**
      * Yes
@@ -618,7 +623,7 @@ interface ModelData {
         inference?: boolean | {
             parameters?: Record<string, unknown>;
         };
-        base_model?: string;
+        base_model?: string | string[];
     };
     /**
      * Library name

package/dist/index.js

@@ -66,6 +66,7 @@ var LIBRARY_TASK_MAPPING_EXCLUDING_TRANSFORMERS = {
   peft: ["text-generation"],
   "pyannote-audio": ["automatic-speech-recognition"],
   "sentence-transformers": ["feature-extraction", "sentence-similarity"],
+  setfit: ["text-classification"],
   sklearn: ["tabular-classification", "tabular-regression", "text-classification"],
   spacy: ["token-classification", "text-classification", "sentence-similarity"],
   "span-marker": ["token-classification"],
@@ -118,7 +119,7 @@ var asteroid = (model) => [
 model = BaseModel.from_pretrained("${model.id}")`
 ];
 function get_base_diffusers_model(model) {
-  return model.cardData?.base_model ?? "fill-in-base-model";
+  return model.cardData?.base_model?.toString() ?? "fill-in-base-model";
 }
 var bertopic = (model) => [
   `from bertopic import BERTopic
@@ -218,8 +219,8 @@ var paddlenlp = (model) => {
       [
         `from paddlenlp.transformers import AutoTokenizer, ${architecture}`,
         "",
-        `tokenizer = AutoTokenizer.from_pretrained("${model.id}"${model.private ? ", use_auth_token=True" : ""}, from_hf_hub=True)`,
-        `model = ${architecture}.from_pretrained("${model.id}"${model.private ? ", use_auth_token=True" : ""}, from_hf_hub=True)`
+        `tokenizer = AutoTokenizer.from_pretrained("${model.id}", from_hf_hub=True)`,
+        `model = ${architecture}.from_pretrained("${model.id}", from_hf_hub=True)`
       ].join("\n")
     ];
   } else {
@@ -228,8 +229,8 @@ var paddlenlp = (model) => {
         `# \u26A0\uFE0F Type of model unknown`,
         `from paddlenlp.transformers import AutoTokenizer, AutoModel`,
         "",
-        `tokenizer = AutoTokenizer.from_pretrained("${model.id}"${model.private ? ", use_auth_token=True" : ""}, from_hf_hub=True)`,
-        `model = AutoModel.from_pretrained("${model.id}"${model.private ? ", use_auth_token=True" : ""}, from_hf_hub=True)`
+        `tokenizer = AutoTokenizer.from_pretrained("${model.id}", from_hf_hub=True)`,
+        `model = AutoModel.from_pretrained("${model.id}", from_hf_hub=True)`
       ].join("\n")
     ];
   }
@@ -365,6 +366,11 @@ var sentenceTransformers = (model) => [
 
 model = SentenceTransformer("${model.id}")`
 ];
+var setfit = (model) => [
+  `from setfit import SetFitModel
+
+model = SetFitModel.from_pretrained("${model.id}")`
+];
 var spacy = (model) => [
   `!pip install https://huggingface.co/${model.id}/resolve/main/${nameWithoutNamespace(model.id)}-any-py3-none-any.whl
 
@@ -637,6 +643,13 @@ var MODEL_LIBRARIES_UI_ELEMENTS = {
     docsUrl: "https://huggingface.co/docs/hub/sentence-transformers",
     snippets: sentenceTransformers
   },
+  setfit: {
+    btnLabel: "setfit",
+    repoName: "setfit",
+    repoUrl: "https://github.com/huggingface/setfit",
+    docsUrl: "https://huggingface.co/docs/hub/setfit",
+    snippets: setfit
+  },
   sklearn: {
     btnLabel: "Scikit-learn",
     repoName: "Scikit-learn",
@@ -2343,6 +2356,10 @@ var taskData5 = {
     {
       description: "A special model for OCR-free Document QA task. Donut model fine-tuned on DocVQA.",
       id: "naver-clova-ix/donut-base-finetuned-docvqa"
+    },
+    {
+      description: "A powerful model for document question answering.",
+      id: "google/pix2struct-docvqa-large"
     }
   ],
   spaces: [
@@ -2353,6 +2370,10 @@ var taskData5 = {
     {
       description: "An application that can answer questions from invoices.",
       id: "impira/invoices"
+    },
+    {
+      description: "An application to compare different document question answering models.",
+      id: "merve/compare_docvqa_models"
     }
   ],
   summary: "Document Question Answering (also known as Document Visual Question Answering) is the task of answering questions on document images. Document question answering models take a (document, question) pair as input and return an answer in natural language. Models usually rely on multi-modal features, combining text, position of words (bounding-boxes) and image.",
@@ -2696,30 +2717,26 @@ var taskData10 = {
   models: [
     {
       description: "A robust image captioning model.",
-      id: "Salesforce/blip-image-captioning-large"
+      id: "Salesforce/blip2-opt-2.7b"
     },
     {
-      description: "A strong image captioning model.",
-      id: "nlpconnect/vit-gpt2-image-captioning"
+      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: "microsoft/trocr-base-printed"
-    },
-    {
-      description: "A strong visual question answering model for scientific diagrams.",
-      id: "google/pix2struct-ai2d-base"
+      id: "facebook/nougat-base"
     },
     {
-      description: "A strong captioning model for UI components.",
-      id: "google/pix2struct-widget-captioning-base"
-    },
-    {
-      description: "A captioning model for images that contain text.",
-      id: "google/pix2struct-textcaps-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"
@@ -2881,7 +2898,6 @@ var taskData12 = {
   ],
   models: [
     {
-      // TO DO: write description
       description: "Solid object detection model trained on the benchmark dataset COCO 2017.",
       id: "facebook/detr-resnet-50"
     },
@@ -2891,9 +2907,13 @@ var taskData12 = {
     }
   ],
   spaces: [
+    {
+      description: "Leaderboard to compare various object detection models across several metrics.",
+      id: "hf-vision/object_detection_leaderboard"
+    },
     {
       description: "An object detection application that can detect unseen objects out of the box.",
-      id: "adirik/OWL-ViT"
+      id: "merve/owlv2"
     },
     {
       description: "An application that contains various object detection models to try from.",
@@ -2939,14 +2959,16 @@ var taskData13 = {
   metrics: [],
   models: [
     {
-      // TO DO: write description
       description: "Strong Depth Estimation model trained on 1.4 million images.",
       id: "Intel/dpt-large"
     },
     {
-      // TO DO: write description
       description: "Strong Depth Estimation model trained on the KITTI dataset.",
-      id: "vinvino02/glpn-kitti"
+      id: "facebook/dpt-dinov2-large-kitti"
+    },
+    {
+      description: "A strong monocular depth estimation model.",
+      id: "Bingxin/Marigold"
     }
   ],
   spaces: [
@@ -3500,12 +3522,12 @@ var taskData22 = {
   ],
   models: [
     {
-      description: "A latent text-to-image diffusion model capable of generating photo-realistic images given any text input.",
-      id: "CompVis/stable-diffusion-v1-4"
+      description: "One of the most powerful image generation models that can generate realistic outputs.",
+      id: "stabilityai/stable-diffusion-xl-base-1.0"
     },
     {
-      description: "A model that can be used to generate images based on text prompts. The DALL\xB7E Mega model is the largest version of DALLE Mini.",
-      id: "dalle-mini/dalle-mega"
+      description: "A powerful yet fast image generation model.",
+      id: "latent-consistency/lcm-lora-sdxl"
     },
     {
       description: "A text-to-image model that can generate coherent text inside image.",
@@ -3522,19 +3544,23 @@ var taskData22 = {
       id: "stabilityai/stable-diffusion"
     },
     {
-      description: "An text-to-image application that can generate coherent text inside the image.",
+      description: "A text-to-image application to generate comics.",
+      id: "jbilcke-hf/ai-comic-factory"
+    },
+    {
+      description: "A text-to-image application that can generate coherent text inside the image.",
       id: "DeepFloyd/IF"
     },
     {
-      description: "An powerful text-to-image application that can generate images.",
-      id: "kakaobrain/karlo"
+      description: "A powerful yet very fast image generation application.",
+      id: "latent-consistency/lcm-lora-for-sdxl"
     },
     {
-      description: "An powerful text-to-image application that can generates 3D representations.",
+      description: "A powerful text-to-image application that can generate 3D representations.",
       id: "hysts/Shap-E"
     },
     {
-      description: "A strong application for `text-to-image`, `image-to-image` and image inpainting.",
+      description: "An application for `text-to-image`, `image-to-image` and image inpainting.",
       id: "ArtGAN/Stable-Diffusion-ControlNet-WebUI"
     }
   ],
@@ -4026,7 +4052,7 @@ var taskData28 = {
   models: [
     {
       description: "A strong model for video generation.",
-      id: "PAIR/text2video-zero-controlnet-canny-arcane"
+      id: "Vchitect/LaVie"
     },
     {
       description: "A robust model for text-to-video generation.",
@@ -4034,7 +4060,7 @@ var taskData28 = {
     },
     {
       description: "A text-to-video generation model with high quality and smooth outputs.",
-      id: "cerspense/zeroscope_v2_576w"
+      id: "hotshotco/Hotshot-XL"
     }
   ],
   spaces: [
@@ -4044,7 +4070,7 @@ var taskData28 = {
     },
     {
       description: "An application that generates video from image and text.",
-      id: "TempoFunk/makeavid-sd-jax"
+      id: "Vchitect/LaVie"
     },
     {
       description: "An application that generates videos from text and provides multi-model support.",
@@ -4274,6 +4300,10 @@ var taskData31 = {
     }
   ],
   spaces: [
+    {
+      description: "An application that compares visual question answering models across different tasks.",
+      id: "merve/pix2struct"
+    },
     {
       description: "An application that can answer questions based on images.",
       id: "nielsr/vilt-vqa"
@@ -4441,8 +4471,8 @@ var TASKS_MODEL_LIBRARIES = {
   "graph-ml": ["transformers"],
   "image-classification": ["keras", "timm", "transformers", "transformers.js"],
   "image-segmentation": ["transformers", "transformers.js"],
-  "image-to-image": ["diffusers", "transformers.js"],
-  "image-to-text": ["transformers.js"],
+  "image-to-image": ["diffusers", "transformers", "transformers.js"],
+  "image-to-text": ["transformers", "transformers.js"],
   "image-to-video": ["diffusers"],
   "video-classification": ["transformers"],
   "mask-generation": ["transformers"],
@@ -4459,7 +4489,7 @@ var TASKS_MODEL_LIBRARIES = {
   "tabular-classification": ["sklearn"],
   "tabular-regression": ["sklearn"],
   "tabular-to-text": ["transformers"],
-  "text-classification": ["adapter-transformers", "spacy", "transformers", "transformers.js"],
+  "text-classification": ["adapter-transformers", "setfit", "spacy", "transformers", "transformers.js"],
   "text-generation": ["transformers", "transformers.js"],
   "text-retrieval": [],
   "text-to-image": ["diffusers"],
@@ -4566,6 +4596,7 @@ var ModelLibrary = /* @__PURE__ */ ((ModelLibrary2) => {
   ModelLibrary2["pyannote-audio"] = "pyannote.audio";
   ModelLibrary2["sample-factory"] = "Sample Factory";
   ModelLibrary2["sentence-transformers"] = "Sentence Transformers";
+  ModelLibrary2["setfit"] = "SetFit";
   ModelLibrary2["sklearn"] = "Scikit-learn";
   ModelLibrary2["spacy"] = "spaCy";
   ModelLibrary2["span-marker"] = "SpanMarker";

package/dist/index.mjs

@@ -28,6 +28,7 @@ var LIBRARY_TASK_MAPPING_EXCLUDING_TRANSFORMERS = {
   peft: ["text-generation"],
   "pyannote-audio": ["automatic-speech-recognition"],
   "sentence-transformers": ["feature-extraction", "sentence-similarity"],
+  setfit: ["text-classification"],
   sklearn: ["tabular-classification", "tabular-regression", "text-classification"],
   spacy: ["token-classification", "text-classification", "sentence-similarity"],
   "span-marker": ["token-classification"],
@@ -80,7 +81,7 @@ var asteroid = (model) => [
 model = BaseModel.from_pretrained("${model.id}")`
 ];
 function get_base_diffusers_model(model) {
-  return model.cardData?.base_model ?? "fill-in-base-model";
+  return model.cardData?.base_model?.toString() ?? "fill-in-base-model";
 }
 var bertopic = (model) => [
   `from bertopic import BERTopic
@@ -180,8 +181,8 @@ var paddlenlp = (model) => {
       [
         `from paddlenlp.transformers import AutoTokenizer, ${architecture}`,
         "",
-        `tokenizer = AutoTokenizer.from_pretrained("${model.id}"${model.private ? ", use_auth_token=True" : ""}, from_hf_hub=True)`,
-        `model = ${architecture}.from_pretrained("${model.id}"${model.private ? ", use_auth_token=True" : ""}, from_hf_hub=True)`
+        `tokenizer = AutoTokenizer.from_pretrained("${model.id}", from_hf_hub=True)`,
+        `model = ${architecture}.from_pretrained("${model.id}", from_hf_hub=True)`
       ].join("\n")
     ];
   } else {
@@ -190,8 +191,8 @@ var paddlenlp = (model) => {
         `# \u26A0\uFE0F Type of model unknown`,
         `from paddlenlp.transformers import AutoTokenizer, AutoModel`,
         "",
-        `tokenizer = AutoTokenizer.from_pretrained("${model.id}"${model.private ? ", use_auth_token=True" : ""}, from_hf_hub=True)`,
-        `model = AutoModel.from_pretrained("${model.id}"${model.private ? ", use_auth_token=True" : ""}, from_hf_hub=True)`
+        `tokenizer = AutoTokenizer.from_pretrained("${model.id}", from_hf_hub=True)`,
+        `model = AutoModel.from_pretrained("${model.id}", from_hf_hub=True)`
       ].join("\n")
     ];
   }
@@ -327,6 +328,11 @@ var sentenceTransformers = (model) => [
 
 model = SentenceTransformer("${model.id}")`
 ];
+var setfit = (model) => [
+  `from setfit import SetFitModel
+
+model = SetFitModel.from_pretrained("${model.id}")`
+];
 var spacy = (model) => [
   `!pip install https://huggingface.co/${model.id}/resolve/main/${nameWithoutNamespace(model.id)}-any-py3-none-any.whl
 
@@ -599,6 +605,13 @@ var MODEL_LIBRARIES_UI_ELEMENTS = {
     docsUrl: "https://huggingface.co/docs/hub/sentence-transformers",
     snippets: sentenceTransformers
   },
+  setfit: {
+    btnLabel: "setfit",
+    repoName: "setfit",
+    repoUrl: "https://github.com/huggingface/setfit",
+    docsUrl: "https://huggingface.co/docs/hub/setfit",
+    snippets: setfit
+  },
   sklearn: {
     btnLabel: "Scikit-learn",
     repoName: "Scikit-learn",
@@ -2305,6 +2318,10 @@ var taskData5 = {
     {
       description: "A special model for OCR-free Document QA task. Donut model fine-tuned on DocVQA.",
       id: "naver-clova-ix/donut-base-finetuned-docvqa"
+    },
+    {
+      description: "A powerful model for document question answering.",
+      id: "google/pix2struct-docvqa-large"
     }
   ],
   spaces: [
@@ -2315,6 +2332,10 @@ var taskData5 = {
     {
       description: "An application that can answer questions from invoices.",
       id: "impira/invoices"
+    },
+    {
+      description: "An application to compare different document question answering models.",
+      id: "merve/compare_docvqa_models"
     }
   ],
   summary: "Document Question Answering (also known as Document Visual Question Answering) is the task of answering questions on document images. Document question answering models take a (document, question) pair as input and return an answer in natural language. Models usually rely on multi-modal features, combining text, position of words (bounding-boxes) and image.",
@@ -2658,30 +2679,26 @@ var taskData10 = {
   models: [
     {
       description: "A robust image captioning model.",
-      id: "Salesforce/blip-image-captioning-large"
+      id: "Salesforce/blip2-opt-2.7b"
     },
     {
-      description: "A strong image captioning model.",
-      id: "nlpconnect/vit-gpt2-image-captioning"
+      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: "microsoft/trocr-base-printed"
-    },
-    {
-      description: "A strong visual question answering model for scientific diagrams.",
-      id: "google/pix2struct-ai2d-base"
+      id: "facebook/nougat-base"
     },
     {
-      description: "A strong captioning model for UI components.",
-      id: "google/pix2struct-widget-captioning-base"
-    },
-    {
-      description: "A captioning model for images that contain text.",
-      id: "google/pix2struct-textcaps-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"
@@ -2843,7 +2860,6 @@ var taskData12 = {
   ],
   models: [
     {
-      // TO DO: write description
       description: "Solid object detection model trained on the benchmark dataset COCO 2017.",
       id: "facebook/detr-resnet-50"
     },
@@ -2853,9 +2869,13 @@ var taskData12 = {
     }
   ],
   spaces: [
+    {
+      description: "Leaderboard to compare various object detection models across several metrics.",
+      id: "hf-vision/object_detection_leaderboard"
+    },
     {
       description: "An object detection application that can detect unseen objects out of the box.",
-      id: "adirik/OWL-ViT"
+      id: "merve/owlv2"
     },
     {
       description: "An application that contains various object detection models to try from.",
@@ -2901,14 +2921,16 @@ var taskData13 = {
   metrics: [],
   models: [
     {
-      // TO DO: write description
       description: "Strong Depth Estimation model trained on 1.4 million images.",
       id: "Intel/dpt-large"
     },
     {
-      // TO DO: write description
       description: "Strong Depth Estimation model trained on the KITTI dataset.",
-      id: "vinvino02/glpn-kitti"
+      id: "facebook/dpt-dinov2-large-kitti"
+    },
+    {
+      description: "A strong monocular depth estimation model.",
+      id: "Bingxin/Marigold"
     }
   ],
   spaces: [
@@ -3462,12 +3484,12 @@ var taskData22 = {
   ],
   models: [
     {
-      description: "A latent text-to-image diffusion model capable of generating photo-realistic images given any text input.",
-      id: "CompVis/stable-diffusion-v1-4"
+      description: "One of the most powerful image generation models that can generate realistic outputs.",
+      id: "stabilityai/stable-diffusion-xl-base-1.0"
     },
     {
-      description: "A model that can be used to generate images based on text prompts. The DALL\xB7E Mega model is the largest version of DALLE Mini.",
-      id: "dalle-mini/dalle-mega"
+      description: "A powerful yet fast image generation model.",
+      id: "latent-consistency/lcm-lora-sdxl"
     },
     {
       description: "A text-to-image model that can generate coherent text inside image.",
@@ -3484,19 +3506,23 @@ var taskData22 = {
       id: "stabilityai/stable-diffusion"
     },
     {
-      description: "An text-to-image application that can generate coherent text inside the image.",
+      description: "A text-to-image application to generate comics.",
+      id: "jbilcke-hf/ai-comic-factory"
+    },
+    {
+      description: "A text-to-image application that can generate coherent text inside the image.",
       id: "DeepFloyd/IF"
     },
     {
-      description: "An powerful text-to-image application that can generate images.",
-      id: "kakaobrain/karlo"
+      description: "A powerful yet very fast image generation application.",
+      id: "latent-consistency/lcm-lora-for-sdxl"
     },
     {
-      description: "An powerful text-to-image application that can generates 3D representations.",
+      description: "A powerful text-to-image application that can generate 3D representations.",
       id: "hysts/Shap-E"
     },
     {
-      description: "A strong application for `text-to-image`, `image-to-image` and image inpainting.",
+      description: "An application for `text-to-image`, `image-to-image` and image inpainting.",
       id: "ArtGAN/Stable-Diffusion-ControlNet-WebUI"
     }
   ],
@@ -3988,7 +4014,7 @@ var taskData28 = {
   models: [
     {
       description: "A strong model for video generation.",
-      id: "PAIR/text2video-zero-controlnet-canny-arcane"
+      id: "Vchitect/LaVie"
     },
     {
       description: "A robust model for text-to-video generation.",
@@ -3996,7 +4022,7 @@ var taskData28 = {
     },
     {
       description: "A text-to-video generation model with high quality and smooth outputs.",
-      id: "cerspense/zeroscope_v2_576w"
+      id: "hotshotco/Hotshot-XL"
     }
   ],
   spaces: [
@@ -4006,7 +4032,7 @@ var taskData28 = {
     },
     {
       description: "An application that generates video from image and text.",
-      id: "TempoFunk/makeavid-sd-jax"
+      id: "Vchitect/LaVie"
     },
     {
       description: "An application that generates videos from text and provides multi-model support.",
@@ -4236,6 +4262,10 @@ var taskData31 = {
     }
   ],
   spaces: [
+    {
+      description: "An application that compares visual question answering models across different tasks.",
+      id: "merve/pix2struct"
+    },
     {
       description: "An application that can answer questions based on images.",
       id: "nielsr/vilt-vqa"
@@ -4403,8 +4433,8 @@ var TASKS_MODEL_LIBRARIES = {
   "graph-ml": ["transformers"],
   "image-classification": ["keras", "timm", "transformers", "transformers.js"],
   "image-segmentation": ["transformers", "transformers.js"],
-  "image-to-image": ["diffusers", "transformers.js"],
-  "image-to-text": ["transformers.js"],
+  "image-to-image": ["diffusers", "transformers", "transformers.js"],
+  "image-to-text": ["transformers", "transformers.js"],
   "image-to-video": ["diffusers"],
   "video-classification": ["transformers"],
   "mask-generation": ["transformers"],
@@ -4421,7 +4451,7 @@ var TASKS_MODEL_LIBRARIES = {
   "tabular-classification": ["sklearn"],
   "tabular-regression": ["sklearn"],
   "tabular-to-text": ["transformers"],
-  "text-classification": ["adapter-transformers", "spacy", "transformers", "transformers.js"],
+  "text-classification": ["adapter-transformers", "setfit", "spacy", "transformers", "transformers.js"],
   "text-generation": ["transformers", "transformers.js"],
   "text-retrieval": [],
   "text-to-image": ["diffusers"],
@@ -4528,6 +4558,7 @@ var ModelLibrary = /* @__PURE__ */ ((ModelLibrary2) => {
   ModelLibrary2["pyannote-audio"] = "pyannote.audio";
   ModelLibrary2["sample-factory"] = "Sample Factory";
   ModelLibrary2["sentence-transformers"] = "Sentence Transformers";
+  ModelLibrary2["setfit"] = "SetFit";
   ModelLibrary2["sklearn"] = "Scikit-learn";
   ModelLibrary2["spacy"] = "spaCy";
   ModelLibrary2["span-marker"] = "SpanMarker";

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@huggingface/tasks",
   "packageManager": "pnpm@8.10.5",
-  "version": "0.0.9",
+  "version": "0.1.1",
   "description": "List of ML tasks for huggingface.co/tasks",
   "repository": "https://github.com/huggingface/huggingface.js.git",
   "publishConfig": {

package/src/default-widget-inputs.ts

@@ -1,4 +1,4 @@
-import type { WidgetExample } from "./model-data";
+import type { WidgetExample } from "./widget-example";
 import type { PipelineType } from "./pipelines";
 
 type LanguageCode = string;

package/src/index.ts

@@ -16,9 +16,8 @@ export {
 } from "./pipelines";
 export { ModelLibrary, ALL_DISPLAY_MODEL_LIBRARY_KEYS } from "./model-libraries";
 export type { ModelLibraryKey } from "./model-libraries";
-export {
-	ModelData,
-	TransformersInfo,
+export type { ModelData, TransformersInfo } from "./model-data";
+export type {
 	WidgetExample,
 	WidgetExampleAttribute,
 	WidgetExampleAssetAndPromptInput,
@@ -37,7 +36,7 @@ export {
 	WidgetExampleOutputLabels,
 	WidgetExampleOutputAnswerScore,
 	WidgetExampleOutputText,
-} from "./model-data";
+} from "./widget-example";
 export { InferenceDisplayability } from "./model-data";
 
 export { TAG_NFAA_CONTENT, OTHER_TAGS_SUGGESTIONS, TAG_TEXT_GENERATION_INFERENCE, TAG_CUSTOM_CODE } from "./tags";

package/src/library-to-tasks.ts

@@ -31,6 +31,7 @@ export const LIBRARY_TASK_MAPPING_EXCLUDING_TRANSFORMERS: Partial<Record<ModelLi
 	peft: ["text-generation"],
 	"pyannote-audio": ["automatic-speech-recognition"],
 	"sentence-transformers": ["feature-extraction", "sentence-similarity"],
+	setfit: ["text-classification"],
 	sklearn: ["tabular-classification", "tabular-regression", "text-classification"],
 	spacy: ["token-classification", "text-classification", "sentence-similarity"],
 	"span-marker": ["token-classification"],

package/src/library-ui-elements.ts

@@ -72,7 +72,7 @@ model = BaseModel.from_pretrained("${model.id}")`,
 ];
 
 function get_base_diffusers_model(model: ModelData): string {
-	return model.cardData?.base_model ?? "fill-in-base-model";
+	return model.cardData?.base_model?.toString() ?? "fill-in-base-model";
 }
 
 const bertopic = (model: ModelData) => [
@@ -187,12 +187,8 @@ const paddlenlp = (model: ModelData) => {
 			[
 				`from paddlenlp.transformers import AutoTokenizer, ${architecture}`,
 				"",
-				`tokenizer = AutoTokenizer.from_pretrained("${model.id}"${
-					model.private ? ", use_auth_token=True" : ""
-				}, from_hf_hub=True)`,
-				`model = ${architecture}.from_pretrained("${model.id}"${
-					model.private ? ", use_auth_token=True" : ""
-				}, from_hf_hub=True)`,
+				`tokenizer = AutoTokenizer.from_pretrained("${model.id}", from_hf_hub=True)`,
+				`model = ${architecture}.from_pretrained("${model.id}", from_hf_hub=True)`,
 			].join("\n"),
 		];
 	} else {
@@ -201,12 +197,8 @@ const paddlenlp = (model: ModelData) => {
 				`# ⚠️ Type of model unknown`,
 				`from paddlenlp.transformers import AutoTokenizer, AutoModel`,
 				"",
-				`tokenizer = AutoTokenizer.from_pretrained("${model.id}"${
-					model.private ? ", use_auth_token=True" : ""
-				}, from_hf_hub=True)`,
-				`model = AutoModel.from_pretrained("${model.id}"${
-					model.private ? ", use_auth_token=True" : ""
-				}, from_hf_hub=True)`,
+				`tokenizer = AutoTokenizer.from_pretrained("${model.id}", from_hf_hub=True)`,
+				`model = AutoModel.from_pretrained("${model.id}", from_hf_hub=True)`,
 			].join("\n"),
 		];
 	}
@@ -358,6 +350,12 @@ const sentenceTransformers = (model: ModelData) => [
 model = SentenceTransformer("${model.id}")`,
 ];
 
+const setfit = (model: ModelData) => [
+	`from setfit import SetFitModel
+
+model = SetFitModel.from_pretrained("${model.id}")`,
+];
+
 const spacy = (model: ModelData) => [
 	`!pip install https://huggingface.co/${model.id}/resolve/main/${nameWithoutNamespace(model.id)}-any-py3-none-any.whl
 
@@ -661,6 +659,13 @@ export const MODEL_LIBRARIES_UI_ELEMENTS: Partial<Record<ModelLibraryKey, Librar
 		docsUrl: "https://huggingface.co/docs/hub/sentence-transformers",
 		snippets: sentenceTransformers,
 	},
+	setfit: {
+		btnLabel: "setfit",
+		repoName: "setfit",
+		repoUrl: "https://github.com/huggingface/setfit",
+		docsUrl: "https://huggingface.co/docs/hub/setfit",
+		snippets: setfit,
+	},
 	sklearn: {
 		btnLabel: "Scikit-learn",
 		repoName: "Scikit-learn",

package/src/model-data.ts

@@ -1,119 +1,5 @@
 import type { PipelineType } from "./pipelines";
-
-type TableData = Record<string, (string | number)[]>;
-
-//#region outputs
-export type WidgetExampleOutputLabels = Array<{ label: string; score: number }>;
-export interface WidgetExampleOutputAnswerScore {
-	answer: string;
-	score: number;
-}
-export interface WidgetExampleOutputText {
-	text: string;
-}
-export interface WidgetExampleOutputUrl {
-	url: string;
-}
-
-export type WidgetExampleOutput =
-	| WidgetExampleOutputLabels
-	| WidgetExampleOutputAnswerScore
-	| WidgetExampleOutputText
-	| WidgetExampleOutputUrl;
-//#endregion
-
-export interface WidgetExampleBase<TOutput> {
-	example_title?: string;
-	group?: string;
-	/**
-	 * Potential overrides to API parameters for this specific example
-	 * (takes precedences over the model card metadata's inference.parameters)
-	 */
-	parameters?: {
-		/// token-classification
-		aggregation_strategy?: string;
-		/// text-generation
-		top_k?: number;
-		top_p?: number;
-		temperature?: number;
-		max_new_tokens?: number;
-		do_sample?: boolean;
-		/// text-to-image
-		negative_prompt?: string;
-		guidance_scale?: number;
-		num_inference_steps?: number;
-	};
-	/**
-	 * Optional output
-	 */
-	output?: TOutput;
-}
-
-export interface WidgetExampleTextInput<TOutput = WidgetExampleOutput> extends WidgetExampleBase<TOutput> {
-	text: string;
-}
-
-export interface WidgetExampleTextAndContextInput<TOutput = WidgetExampleOutput>
-	extends WidgetExampleTextInput<TOutput> {
-	context: string;
-}
-
-export interface WidgetExampleTextAndTableInput<TOutput = WidgetExampleOutput> extends WidgetExampleTextInput<TOutput> {
-	table: TableData;
-}
-
-export interface WidgetExampleAssetInput<TOutput = WidgetExampleOutput> extends WidgetExampleBase<TOutput> {
-	src: string;
-}
-export interface WidgetExampleAssetAndPromptInput<TOutput = WidgetExampleOutput>
-	extends WidgetExampleAssetInput<TOutput> {
-	prompt: string;
-}
-
-export type WidgetExampleAssetAndTextInput<TOutput = WidgetExampleOutput> = WidgetExampleAssetInput<TOutput> &
-	WidgetExampleTextInput<TOutput>;
-
-export type WidgetExampleAssetAndZeroShotInput<TOutput = WidgetExampleOutput> = WidgetExampleAssetInput<TOutput> &
-	WidgetExampleZeroShotTextInput<TOutput>;
-
-export interface WidgetExampleStructuredDataInput<TOutput = WidgetExampleOutput> extends WidgetExampleBase<TOutput> {
-	structured_data: TableData;
-}
-
-export interface WidgetExampleTableDataInput<TOutput = WidgetExampleOutput> extends WidgetExampleBase<TOutput> {
-	table: TableData;
-}
-
-export interface WidgetExampleZeroShotTextInput<TOutput = WidgetExampleOutput> extends WidgetExampleTextInput<TOutput> {
-	text: string;
-	candidate_labels: string;
-	multi_class: boolean;
-}
-
-export interface WidgetExampleSentenceSimilarityInput<TOutput = WidgetExampleOutput>
-	extends WidgetExampleBase<TOutput> {
-	source_sentence: string;
-	sentences: string[];
-}
-
-//#endregion
-
-export type WidgetExample<TOutput = WidgetExampleOutput> =
-	| WidgetExampleTextInput<TOutput>
-	| WidgetExampleTextAndContextInput<TOutput>
-	| WidgetExampleTextAndTableInput<TOutput>
-	| WidgetExampleAssetInput<TOutput>
-	| WidgetExampleAssetAndPromptInput<TOutput>
-	| WidgetExampleAssetAndTextInput<TOutput>
-	| WidgetExampleAssetAndZeroShotInput<TOutput>
-	| WidgetExampleStructuredDataInput<TOutput>
-	| WidgetExampleTableDataInput<TOutput>
-	| WidgetExampleZeroShotTextInput<TOutput>
-	| WidgetExampleSentenceSimilarityInput<TOutput>;
-
-type KeysOfUnion<T> = T extends unknown ? keyof T : never;
-
-export type WidgetExampleAttribute = KeysOfUnion<WidgetExample>;
+import type { WidgetExample } from "./widget-example";
 
 export enum InferenceDisplayability {
 	/**
@@ -207,7 +93,7 @@ export interface ModelData {
 			| {
 					parameters?: Record<string, unknown>;
 			  };
-		base_model?: string;
+		base_model?: string | string[];
 	};
 	/**
 	 * Library name

package/src/model-libraries.ts

@@ -23,6 +23,7 @@ export enum ModelLibrary {
 	"pyannote-audio" = "pyannote.audio",
 	"sample-factory" = "Sample Factory",
 	"sentence-transformers" = "Sentence Transformers",
+	"setfit" = "SetFit",
 	"sklearn" = "Scikit-learn",
 	"spacy" = "spaCy",
 	"span-marker" = "SpanMarker",

package/src/tasks/depth-estimation/data.ts

@@ -24,14 +24,16 @@ const taskData: TaskDataCustom = {
 	metrics: [],
 	models: [
 		{
-			// TO DO: write description
 			description: "Strong Depth Estimation model trained on 1.4 million images.",
 			id: "Intel/dpt-large",
 		},
 		{
-			// TO DO: write description
 			description: "Strong Depth Estimation model trained on the KITTI dataset.",
-			id: "vinvino02/glpn-kitti",
+			id: "facebook/dpt-dinov2-large-kitti",
+		},
+		{
+			description: "A strong monocular depth estimation model.",
+			id: "Bingxin/Marigold",
 		},
 	],
 	spaces: [

package/src/tasks/document-question-answering/data.ts

@@ -50,6 +50,10 @@ const taskData: TaskDataCustom = {
 			description: "A special model for OCR-free Document QA task. Donut model fine-tuned on DocVQA.",
 			id: "naver-clova-ix/donut-base-finetuned-docvqa",
 		},
+		{
+			description: "A powerful model for document question answering.",
+			id: "google/pix2struct-docvqa-large",
+		},
 	],
 	spaces: [
 		{
@@ -60,6 +64,10 @@ const taskData: TaskDataCustom = {
 			description: "An application that can answer questions from invoices.",
 			id: "impira/invoices",
 		},
+		{
+			description: "An application to compare different document question answering models.",
+			id: "merve/compare_docvqa_models",
+		},
 	],
 	summary:
 		"Document Question Answering (also known as Document Visual Question Answering) is the task of answering questions on document images. Document question answering models take a (document, question) pair as input and return an answer in natural language. Models usually rely on multi-modal features, combining text, position of words (bounding-boxes) and image.",

package/src/tasks/image-to-text/about.md

@@ -27,6 +27,19 @@ captioner("https://huggingface.co/datasets/Narsil/image_dummy/resolve/main/parro
 ## [{'generated_text': 'two birds are standing next to each other '}]
 ```
 
+### Conversation about the Image
+
+Some text generation models also take image inputs. These are called vision language models. You can use `image-to-text` pipeline to use these models like below.
+
+```python
+from transformers import pipeline
+
+mm_pipeline = pipeline("image-to-text",model="llava-hf/llava-1.5-7b-hf")
+mm_pipeline("https://huggingface.co/spaces/llava-hf/llava-4bit/resolve/main/examples/baklava.png", "How to make this pastry?")
+
+## [{'generated_text': 'To create these pastries, you will need a few key ingredients and tools. Firstly, gather the dough by combining flour with water in your mixing bowl until it forms into an elastic ball that can be easily rolled out on top of another surface or table without breaking apart (like pizza).'}]
+```
+
 ### OCR
 
 This code snippet uses Microsoft’s TrOCR, an encoder-decoder model consisting of an image Transformer encoder and a text Transformer decoder for state-of-the-art optical character recognition (OCR) on single-text line images.

package/src/tasks/image-to-text/data.ts

@@ -32,30 +32,26 @@ const taskData: TaskDataCustom = {
 	models: [
 		{
 			description: "A robust image captioning model.",
-			id: "Salesforce/blip-image-captioning-large",
+			id: "Salesforce/blip2-opt-2.7b",
 		},
 		{
-			description: "A strong image captioning model.",
-			id: "nlpconnect/vit-gpt2-image-captioning",
+			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: "microsoft/trocr-base-printed",
+			id: "facebook/nougat-base",
 		},
 		{
-			description: "A strong visual question answering model for scientific diagrams.",
-			id: "google/pix2struct-ai2d-base",
-		},
-		{
-			description: "A strong captioning model for UI components.",
-			id: "google/pix2struct-widget-captioning-base",
-		},
-		{
-			description: "A captioning model for images that contain text.",
-			id: "google/pix2struct-textcaps-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",

package/src/tasks/index.ts

@@ -51,8 +51,8 @@ export const TASKS_MODEL_LIBRARIES: Record<PipelineType, ModelLibraryKey[]> = {
 	"graph-ml": ["transformers"],
 	"image-classification": ["keras", "timm", "transformers", "transformers.js"],
 	"image-segmentation": ["transformers", "transformers.js"],
-	"image-to-image": ["diffusers", "transformers.js"],
-	"image-to-text": ["transformers.js"],
+	"image-to-image": ["diffusers", "transformers", "transformers.js"],
+	"image-to-text": ["transformers", "transformers.js"],
 	"image-to-video": ["diffusers"],
 	"video-classification": ["transformers"],
 	"mask-generation": ["transformers"],
@@ -69,7 +69,7 @@ export const TASKS_MODEL_LIBRARIES: Record<PipelineType, ModelLibraryKey[]> = {
 	"tabular-classification": ["sklearn"],
 	"tabular-regression": ["sklearn"],
 	"tabular-to-text": ["transformers"],
-	"text-classification": ["adapter-transformers", "spacy", "transformers", "transformers.js"],
+	"text-classification": ["adapter-transformers", "setfit", "spacy", "transformers", "transformers.js"],
 	"text-generation": ["transformers", "transformers.js"],
 	"text-retrieval": [],
 	"text-to-image": ["diffusers"],

package/src/tasks/object-detection/data.ts

@@ -40,7 +40,6 @@ const taskData: TaskDataCustom = {
 	],
 	models: [
 		{
-			// TO DO: write description
 			description: "Solid object detection model trained on the benchmark dataset COCO 2017.",
 			id: "facebook/detr-resnet-50",
 		},
@@ -50,9 +49,13 @@ const taskData: TaskDataCustom = {
 		},
 	],
 	spaces: [
+		{
+			description: "Leaderboard to compare various object detection models across several metrics.",
+			id: "hf-vision/object_detection_leaderboard",
+		},
 		{
 			description: "An object detection application that can detect unseen objects out of the box.",
-			id: "adirik/OWL-ViT",
+			id: "merve/owlv2",
 		},
 		{
 			description: "An application that contains various object detection models to try from.",

package/src/tasks/table-question-answering/about.md

@@ -31,7 +31,7 @@ tqa = pipeline(task="table-question-answering", model="google/tapas-large-finetu
 
 # result
 
-print(tqa(table=table, query=query)['cells'][0])
+print(tqa(table=table, query=question)['cells'][0])
 #53
 
 ```

package/src/tasks/text-generation/about.md

@@ -42,6 +42,10 @@ When it comes to text generation, the underlying language model can come in seve
 
 - **Human feedback models:** these models extend base and instruction-trained models by incorporating human feedback that rates the quality of the generated text according to criteria like [helpfulness, honesty, and harmlessness](https://arxiv.org/abs/2112.00861). The human feedback is then combined with an optimization technique like reinforcement learning to align the original model to be closer with human preferences. The overall methodology is often called [Reinforcement Learning from Human Feedback](https://huggingface.co/blog/rlhf), or RLHF for short. [Llama2-Chat](https://huggingface.co/meta-llama/Llama-2-7b-chat-hf) is an open-source model aligned through human feedback.
 
+## Text Generation from Image and Text
+
+There are language models that can input both text and image and output text, called vision language models. [LLaVA](https://huggingface.co/llava-hf/llava-1.5-7b-hf) and [BLIP-2](https://huggingface.co/Salesforce/blip2-opt-2.7b) are good examples. Although they work just like other language models by means of input parameters for generation, since they also take input images, you can use them with `image-to-text` pipeline. You can find information about the pipeline in [image-to-text](https://huggingface.co/tasks/image-to-text) task page.
+
 ## Inference
 
 You can use the 🤗 Transformers library `text-generation` pipeline to do inference with Text Generation models. It takes an incomplete text and returns multiple outputs with which the text can be completed.

package/src/tasks/text-to-image/data.ts

@@ -45,14 +45,12 @@ const taskData: TaskDataCustom = {
 	],
 	models: [
 		{
-			description:
-				"A latent text-to-image diffusion model capable of generating photo-realistic images given any text input.",
-			id: "CompVis/stable-diffusion-v1-4",
+			description: "One of the most powerful image generation models that can generate realistic outputs.",
+			id: "stabilityai/stable-diffusion-xl-base-1.0",
 		},
 		{
-			description:
-				"A model that can be used to generate images based on text prompts. The DALL·E Mega model is the largest version of DALLE Mini.",
-			id: "dalle-mini/dalle-mega",
+			description: "A powerful yet fast image generation model.",
+			id: "latent-consistency/lcm-lora-sdxl",
 		},
 		{
 			description: "A text-to-image model that can generate coherent text inside image.",
@@ -69,19 +67,23 @@ const taskData: TaskDataCustom = {
 			id: "stabilityai/stable-diffusion",
 		},
 		{
-			description: "An text-to-image application that can generate coherent text inside the image.",
+			description: "A text-to-image application to generate comics.",
+			id: "jbilcke-hf/ai-comic-factory",
+		},
+		{
+			description: "A text-to-image application that can generate coherent text inside the image.",
 			id: "DeepFloyd/IF",
 		},
 		{
-			description: "An powerful text-to-image application that can generate images.",
-			id: "kakaobrain/karlo",
+			description: "A powerful yet very fast image generation application.",
+			id: "latent-consistency/lcm-lora-for-sdxl",
 		},
 		{
-			description: "An powerful text-to-image application that can generates 3D representations.",
+			description: "A powerful text-to-image application that can generate 3D representations.",
 			id: "hysts/Shap-E",
 		},
 		{
-			description: "A strong application for `text-to-image`, `image-to-image` and image inpainting.",
+			description: "An application for `text-to-image`, `image-to-image` and image inpainting.",
 			id: "ArtGAN/Stable-Diffusion-ControlNet-WebUI",
 		},
 	],

package/src/tasks/text-to-video/data.ts

@@ -68,7 +68,7 @@ const taskData: TaskDataCustom = {
 	models: [
 		{
 			description: "A strong model for video generation.",
-			id: "PAIR/text2video-zero-controlnet-canny-arcane",
+			id: "Vchitect/LaVie",
 		},
 		{
 			description: "A robust model for text-to-video generation.",
@@ -76,7 +76,7 @@ const taskData: TaskDataCustom = {
 		},
 		{
 			description: "A text-to-video generation model with high quality and smooth outputs.",
-			id: "cerspense/zeroscope_v2_576w",
+			id: "hotshotco/Hotshot-XL",
 		},
 	],
 	spaces: [
@@ -86,7 +86,7 @@ const taskData: TaskDataCustom = {
 		},
 		{
 			description: "An application that generates video from image and text.",
-			id: "TempoFunk/makeavid-sd-jax",
+			id: "Vchitect/LaVie",
 		},
 		{
 			description: "An application that generates videos from text and provides multi-model support.",

package/src/tasks/visual-question-answering/data.ts

@@ -71,6 +71,10 @@ const taskData: TaskDataCustom = {
 		},
 	],
 	spaces: [
+		{
+			description: "An application that compares visual question answering models across different tasks.",
+			id: "merve/pix2struct",
+		},
 		{
 			description: "An application that can answer questions based on images.",
 			id: "nielsr/vilt-vqa",

package/src/widget-example.ts

@@ -0,0 +1,118 @@
+/**
+ * See default-widget-inputs.ts for the default widget inputs, this files only contains the types
+ */
+
+type TableData = Record<string, (string | number)[]>;
+
+//#region outputs
+export type WidgetExampleOutputLabels = Array<{ label: string; score: number }>;
+export interface WidgetExampleOutputAnswerScore {
+	answer: string;
+	score: number;
+}
+export interface WidgetExampleOutputText {
+	text: string;
+}
+export interface WidgetExampleOutputUrl {
+	url: string;
+}
+
+export type WidgetExampleOutput =
+	| WidgetExampleOutputLabels
+	| WidgetExampleOutputAnswerScore
+	| WidgetExampleOutputText
+	| WidgetExampleOutputUrl;
+//#endregion
+
+export interface WidgetExampleBase<TOutput> {
+	example_title?: string;
+	group?: string;
+	/**
+	 * Potential overrides to API parameters for this specific example
+	 * (takes precedences over the model card metadata's inference.parameters)
+	 */
+	parameters?: {
+		/// token-classification
+		aggregation_strategy?: string;
+		/// text-generation
+		top_k?: number;
+		top_p?: number;
+		temperature?: number;
+		max_new_tokens?: number;
+		do_sample?: boolean;
+		/// text-to-image
+		negative_prompt?: string;
+		guidance_scale?: number;
+		num_inference_steps?: number;
+	};
+	/**
+	 * Optional output
+	 */
+	output?: TOutput;
+}
+
+export interface WidgetExampleTextInput<TOutput = WidgetExampleOutput> extends WidgetExampleBase<TOutput> {
+	text: string;
+}
+
+export interface WidgetExampleTextAndContextInput<TOutput = WidgetExampleOutput>
+	extends WidgetExampleTextInput<TOutput> {
+	context: string;
+}
+
+export interface WidgetExampleTextAndTableInput<TOutput = WidgetExampleOutput> extends WidgetExampleTextInput<TOutput> {
+	table: TableData;
+}
+
+export interface WidgetExampleAssetInput<TOutput = WidgetExampleOutput> extends WidgetExampleBase<TOutput> {
+	src: string;
+}
+export interface WidgetExampleAssetAndPromptInput<TOutput = WidgetExampleOutput>
+	extends WidgetExampleAssetInput<TOutput> {
+	prompt: string;
+}
+
+export type WidgetExampleAssetAndTextInput<TOutput = WidgetExampleOutput> = WidgetExampleAssetInput<TOutput> &
+	WidgetExampleTextInput<TOutput>;
+
+export type WidgetExampleAssetAndZeroShotInput<TOutput = WidgetExampleOutput> = WidgetExampleAssetInput<TOutput> &
+	WidgetExampleZeroShotTextInput<TOutput>;
+
+export interface WidgetExampleStructuredDataInput<TOutput = WidgetExampleOutput> extends WidgetExampleBase<TOutput> {
+	structured_data: TableData;
+}
+
+export interface WidgetExampleTableDataInput<TOutput = WidgetExampleOutput> extends WidgetExampleBase<TOutput> {
+	table: TableData;
+}
+
+export interface WidgetExampleZeroShotTextInput<TOutput = WidgetExampleOutput> extends WidgetExampleTextInput<TOutput> {
+	text: string;
+	candidate_labels: string;
+	multi_class: boolean;
+}
+
+export interface WidgetExampleSentenceSimilarityInput<TOutput = WidgetExampleOutput>
+	extends WidgetExampleBase<TOutput> {
+	source_sentence: string;
+	sentences: string[];
+}
+
+//#endregion
+
+export type WidgetExample<TOutput = WidgetExampleOutput> =
+	| WidgetExampleTextInput<TOutput>
+	| WidgetExampleTextAndContextInput<TOutput>
+	| WidgetExampleTextAndTableInput<TOutput>
+	| WidgetExampleAssetInput<TOutput>
+	| WidgetExampleAssetAndPromptInput<TOutput>
+	| WidgetExampleAssetAndTextInput<TOutput>
+	| WidgetExampleAssetAndZeroShotInput<TOutput>
+	| WidgetExampleStructuredDataInput<TOutput>
+	| WidgetExampleTableDataInput<TOutput>
+	| WidgetExampleZeroShotTextInput<TOutput>
+	| WidgetExampleSentenceSimilarityInput<TOutput>;
+
+type KeysOfUnion<T> = T extends unknown ? keyof T : never;
+
+export type WidgetExampleAttribute = KeysOfUnion<WidgetExample>;