@huggingface/tasks 0.1.3 → 0.2.0

package/dist/index.d.ts

@@ -42,7 +42,7 @@ declare enum ModelLibrary {
     "mindspore" = "MindSpore"
 }
 type ModelLibraryKey = keyof typeof ModelLibrary;
-declare const ALL_DISPLAY_MODEL_LIBRARY_KEYS: string[];
+declare const ALL_DISPLAY_MODEL_LIBRARY_KEYS: ("speechbrain" | "fastai" | "adapter-transformers" | "allennlp" | "asteroid" | "bertopic" | "diffusers" | "doctr" | "espnet" | "fairseq" | "flair" | "keras" | "k2" | "mlx" | "nemo" | "open_clip" | "paddlenlp" | "peft" | "pyannote-audio" | "sample-factory" | "sentence-transformers" | "setfit" | "sklearn" | "spacy" | "span-marker" | "tensorflowtts" | "timm" | "transformers" | "transformers.js" | "stanza" | "fasttext" | "stable-baselines3" | "ml-agents" | "pythae" | "mindspore")[];
 
 declare const MODALITIES: readonly ["cv", "nlp", "audio", "tabular", "multimodal", "rl", "other"];
 type Modality = (typeof MODALITIES)[number];
@@ -742,11 +742,6 @@ interface TaskData {
 }
 type TaskDataCustom = Omit<TaskData, "id" | "label" | "libraries">;
 
-declare const TAG_NFAA_CONTENT = "not-for-all-audiences";
-declare const OTHER_TAGS_SUGGESTIONS: string[];
-declare const TAG_TEXT_GENERATION_INFERENCE = "text-generation-inference";
-declare const TAG_CUSTOM_CODE = "custom_code";
-
 declare function getModelInputSnippet(model: ModelData, noWrap?: boolean, noQuotes?: boolean): string;
 
 declare const inputs_getModelInputSnippet: typeof getModelInputSnippet;
@@ -854,4 +849,4 @@ declare namespace index {
   };
 }
 
-export { ALL_DISPLAY_MODEL_LIBRARY_KEYS, ExampleRepo, InferenceDisplayability, LIBRARY_TASK_MAPPING_EXCLUDING_TRANSFORMERS, LibraryUiElement, MAPPING_DEFAULT_WIDGET, MODALITIES, MODALITY_LABELS, MODEL_LIBRARIES_UI_ELEMENTS, Modality, ModelData, ModelLibrary, ModelLibraryKey, OTHER_TAGS_SUGGESTIONS, PIPELINE_DATA, PIPELINE_TYPES, PIPELINE_TYPES_SET, PipelineData, PipelineType, SUBTASK_TYPES, TAG_CUSTOM_CODE, TAG_NFAA_CONTENT, TAG_TEXT_GENERATION_INFERENCE, TASKS_DATA, TASKS_MODEL_LIBRARIES, TaskData, TaskDataCustom, TaskDemo, TaskDemoEntry, TransformersInfo, WidgetExample, WidgetExampleAssetAndPromptInput, WidgetExampleAssetAndTextInput, WidgetExampleAssetAndZeroShotInput, WidgetExampleAssetInput, WidgetExampleAttribute, WidgetExampleOutput, WidgetExampleOutputAnswerScore, WidgetExampleOutputLabels, WidgetExampleOutputText, WidgetExampleOutputUrl, WidgetExampleSentenceSimilarityInput, WidgetExampleStructuredDataInput, WidgetExampleTableDataInput, WidgetExampleTextAndContextInput, WidgetExampleTextAndTableInput, WidgetExampleTextInput, WidgetExampleZeroShotTextInput, index as snippets };
+export { ALL_DISPLAY_MODEL_LIBRARY_KEYS, ExampleRepo, InferenceDisplayability, LIBRARY_TASK_MAPPING_EXCLUDING_TRANSFORMERS, LibraryUiElement, MAPPING_DEFAULT_WIDGET, MODALITIES, MODALITY_LABELS, MODEL_LIBRARIES_UI_ELEMENTS, Modality, ModelData, ModelLibrary, ModelLibraryKey, PIPELINE_DATA, PIPELINE_TYPES, PIPELINE_TYPES_SET, PipelineData, PipelineType, SUBTASK_TYPES, TASKS_DATA, TASKS_MODEL_LIBRARIES, TaskData, TaskDataCustom, TaskDemo, TaskDemoEntry, TransformersInfo, WidgetExample, WidgetExampleAssetAndPromptInput, WidgetExampleAssetAndTextInput, WidgetExampleAssetAndZeroShotInput, WidgetExampleAssetInput, WidgetExampleAttribute, WidgetExampleOutput, WidgetExampleOutputAnswerScore, WidgetExampleOutputLabels, WidgetExampleOutputText, WidgetExampleOutputUrl, WidgetExampleSentenceSimilarityInput, WidgetExampleStructuredDataInput, WidgetExampleTableDataInput, WidgetExampleTextAndContextInput, WidgetExampleTextAndTableInput, WidgetExampleTextInput, WidgetExampleZeroShotTextInput, index as snippets };

package/dist/index.js

@@ -28,14 +28,10 @@ __export(src_exports, {
   MODALITY_LABELS: () => MODALITY_LABELS,
   MODEL_LIBRARIES_UI_ELEMENTS: () => MODEL_LIBRARIES_UI_ELEMENTS,
   ModelLibrary: () => ModelLibrary,
-  OTHER_TAGS_SUGGESTIONS: () => OTHER_TAGS_SUGGESTIONS,
   PIPELINE_DATA: () => PIPELINE_DATA,
   PIPELINE_TYPES: () => PIPELINE_TYPES,
   PIPELINE_TYPES_SET: () => PIPELINE_TYPES_SET,
   SUBTASK_TYPES: () => SUBTASK_TYPES,
-  TAG_CUSTOM_CODE: () => TAG_CUSTOM_CODE,
-  TAG_NFAA_CONTENT: () => TAG_NFAA_CONTENT,
-  TAG_TEXT_GENERATION_INFERENCE: () => TAG_TEXT_GENERATION_INFERENCE,
   TASKS_DATA: () => TASKS_DATA,
   TASKS_MODEL_LIBRARIES: () => TASKS_MODEL_LIBRARIES,
   snippets: () => snippets_exports
@@ -4627,7 +4623,8 @@ var ModelLibrary = /* @__PURE__ */ ((ModelLibrary2) => {
   ModelLibrary2["mindspore"] = "MindSpore";
   return ModelLibrary2;
 })(ModelLibrary || {});
-var ALL_DISPLAY_MODEL_LIBRARY_KEYS = Object.keys(ModelLibrary).filter(
+var ALL_MODEL_LIBRARY_KEYS = Object.keys(ModelLibrary);
+var ALL_DISPLAY_MODEL_LIBRARY_KEYS = ALL_MODEL_LIBRARY_KEYS.filter(
   (k) => !["doctr", "k2", "mindspore", "tensorflowtts"].includes(k)
 );
 
@@ -4642,23 +4639,6 @@ var InferenceDisplayability = /* @__PURE__ */ ((InferenceDisplayability2) => {
   return InferenceDisplayability2;
 })(InferenceDisplayability || {});
 
-// src/tags.ts
-var TAG_NFAA_CONTENT = "not-for-all-audiences";
-var OTHER_TAGS_SUGGESTIONS = [
-  "chemistry",
-  "biology",
-  "finance",
-  "legal",
-  "music",
-  "art",
-  "code",
-  "climate",
-  "medical",
-  TAG_NFAA_CONTENT
-];
-var TAG_TEXT_GENERATION_INFERENCE = "text-generation-inference";
-var TAG_CUSTOM_CODE = "custom_code";
-
 // src/snippets/index.ts
 var snippets_exports = {};
 __export(snippets_exports, {
@@ -5130,14 +5110,10 @@ function hasJsInferenceSnippet(model) {
   MODALITY_LABELS,
   MODEL_LIBRARIES_UI_ELEMENTS,
   ModelLibrary,
-  OTHER_TAGS_SUGGESTIONS,
   PIPELINE_DATA,
   PIPELINE_TYPES,
   PIPELINE_TYPES_SET,
   SUBTASK_TYPES,
-  TAG_CUSTOM_CODE,
-  TAG_NFAA_CONTENT,
-  TAG_TEXT_GENERATION_INFERENCE,
   TASKS_DATA,
   TASKS_MODEL_LIBRARIES,
   snippets

package/dist/index.mjs

@@ -4589,7 +4589,8 @@ var ModelLibrary = /* @__PURE__ */ ((ModelLibrary2) => {
   ModelLibrary2["mindspore"] = "MindSpore";
   return ModelLibrary2;
 })(ModelLibrary || {});
-var ALL_DISPLAY_MODEL_LIBRARY_KEYS = Object.keys(ModelLibrary).filter(
+var ALL_MODEL_LIBRARY_KEYS = Object.keys(ModelLibrary);
+var ALL_DISPLAY_MODEL_LIBRARY_KEYS = ALL_MODEL_LIBRARY_KEYS.filter(
   (k) => !["doctr", "k2", "mindspore", "tensorflowtts"].includes(k)
 );
 
@@ -4604,23 +4605,6 @@ var InferenceDisplayability = /* @__PURE__ */ ((InferenceDisplayability2) => {
   return InferenceDisplayability2;
 })(InferenceDisplayability || {});
 
-// src/tags.ts
-var TAG_NFAA_CONTENT = "not-for-all-audiences";
-var OTHER_TAGS_SUGGESTIONS = [
-  "chemistry",
-  "biology",
-  "finance",
-  "legal",
-  "music",
-  "art",
-  "code",
-  "climate",
-  "medical",
-  TAG_NFAA_CONTENT
-];
-var TAG_TEXT_GENERATION_INFERENCE = "text-generation-inference";
-var TAG_CUSTOM_CODE = "custom_code";
-
 // src/snippets/index.ts
 var snippets_exports = {};
 __export(snippets_exports, {
@@ -5091,14 +5075,10 @@ export {
   MODALITY_LABELS,
   MODEL_LIBRARIES_UI_ELEMENTS,
   ModelLibrary,
-  OTHER_TAGS_SUGGESTIONS,
   PIPELINE_DATA,
   PIPELINE_TYPES,
   PIPELINE_TYPES_SET,
   SUBTASK_TYPES,
-  TAG_CUSTOM_CODE,
-  TAG_NFAA_CONTENT,
-  TAG_TEXT_GENERATION_INFERENCE,
   TASKS_DATA,
   TASKS_MODEL_LIBRARIES,
   snippets_exports as snippets

package/package.json

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

package/src/index.ts

@@ -39,8 +39,6 @@ export type {
 } from "./widget-example";
 export { InferenceDisplayability } from "./model-data";
 
-export { TAG_NFAA_CONTENT, OTHER_TAGS_SUGGESTIONS, TAG_TEXT_GENERATION_INFERENCE, TAG_CUSTOM_CODE } from "./tags";
-
 import * as snippets from "./snippets";
 export { snippets };
 

package/src/model-libraries.ts

@@ -44,6 +44,8 @@ export enum ModelLibrary {
 
 export type ModelLibraryKey = keyof typeof ModelLibrary;
 
-export const ALL_DISPLAY_MODEL_LIBRARY_KEYS = Object.keys(ModelLibrary).filter(
+export const ALL_MODEL_LIBRARY_KEYS = Object.keys(ModelLibrary) as ModelLibraryKey[];
+
+export const ALL_DISPLAY_MODEL_LIBRARY_KEYS = ALL_MODEL_LIBRARY_KEYS.filter(
 	(k) => !["doctr", "k2", "mindspore", "tensorflowtts"].includes(k)
 );

package/src/tags.ts

@@ -1,15 +0,0 @@
-export const TAG_NFAA_CONTENT = "not-for-all-audiences";
-export const OTHER_TAGS_SUGGESTIONS = [
-	"chemistry",
-	"biology",
-	"finance",
-	"legal",
-	"music",
-	"art",
-	"code",
-	"climate",
-	"medical",
-	TAG_NFAA_CONTENT,
-];
-export const TAG_TEXT_GENERATION_INFERENCE = "text-generation-inference";
-export const TAG_CUSTOM_CODE = "custom_code";