@huggingface/tasks 0.5.2 → 0.6.0

package/dist/index.cjs

@@ -4083,6 +4083,11 @@ var flair = (model) => [
 
 tagger = SequenceTagger.load("${model.id}")`
 ];
+var gliner = (model) => [
+  `from model import GLiNER
+
+model = GLiNER.from_pretrained("${model.id}")`
+];
 var keras = (model) => [
   `from huggingface_hub import from_pretrained_keras
 
@@ -4578,6 +4583,16 @@ var MODEL_LIBRARIES_UI_ELEMENTS = {
       term: { path: "pytorch_model.bin" }
     }
   },
+  gliner: {
+    prettyLabel: "GLiNER",
+    repoName: "GLiNER",
+    repoUrl: "https://github.com/urchade/GLiNER",
+    snippets: gliner,
+    filter: false,
+    countDownloads: {
+      term: { path: "gliner_config.json" }
+    }
+  },
   keras: {
     prettyLabel: "Keras",
     repoName: "Keras",
@@ -4793,6 +4808,14 @@ var MODEL_LIBRARIES_UI_ELEMENTS = {
     countDownloads: {
       wildcard: { path: "*.sentis" }
     }
+  },
+  whisperkit: {
+    prettyLabel: "WhisperKit",
+    repoName: "WhisperKit",
+    repoUrl: "https://github.com/argmaxinc/WhisperKit",
+    countDownloads: {
+      wildcard: { path: "*/model.mil" }
+    }
   }
 };
 var ALL_MODEL_LIBRARY_KEYS = Object.keys(MODEL_LIBRARIES_UI_ELEMENTS);

package/dist/index.d.ts

@@ -839,6 +839,18 @@ declare const MODEL_LIBRARIES_UI_ELEMENTS: {
             };
         };
     };
+    gliner: {
+        prettyLabel: string;
+        repoName: string;
+        repoUrl: string;
+        snippets: (model: ModelData) => string[];
+        filter: false;
+        countDownloads: {
+            term: {
+                path: string;
+            };
+        };
+    };
     keras: {
         prettyLabel: string;
         repoName: string;
@@ -1093,10 +1105,20 @@ declare const MODEL_LIBRARIES_UI_ELEMENTS: {
             };
         };
     };
+    whisperkit: {
+        prettyLabel: string;
+        repoName: string;
+        repoUrl: string;
+        countDownloads: {
+            wildcard: {
+                path: string;
+            };
+        };
+    };
 };
 type ModelLibraryKey = keyof typeof MODEL_LIBRARIES_UI_ELEMENTS;
-declare const ALL_MODEL_LIBRARY_KEYS: ("sklearn" | "adapter-transformers" | "allennlp" | "asteroid" | "audiocraft" | "bertopic" | "diffusers" | "doctr" | "espnet" | "fairseq" | "fastai" | "fasttext" | "flair" | "keras" | "k2" | "mindspore" | "ml-agents" | "mlx" | "nemo" | "open_clip" | "paddlenlp" | "peft" | "pyannote-audio" | "pythae" | "sample-factory" | "sentence-transformers" | "setfit" | "spacy" | "span-marker" | "speechbrain" | "stable-baselines3" | "stanza" | "tensorflowtts" | "timm" | "transformers" | "transformers.js" | "unity-sentis")[];
-declare const ALL_DISPLAY_MODEL_LIBRARY_KEYS: ("sklearn" | "adapter-transformers" | "allennlp" | "asteroid" | "audiocraft" | "bertopic" | "diffusers" | "doctr" | "espnet" | "fairseq" | "fastai" | "fasttext" | "flair" | "keras" | "k2" | "mindspore" | "ml-agents" | "mlx" | "nemo" | "open_clip" | "paddlenlp" | "peft" | "pyannote-audio" | "pythae" | "sample-factory" | "sentence-transformers" | "setfit" | "spacy" | "span-marker" | "speechbrain" | "stable-baselines3" | "stanza" | "tensorflowtts" | "timm" | "transformers" | "transformers.js" | "unity-sentis")[];
+declare const ALL_MODEL_LIBRARY_KEYS: ("sklearn" | "adapter-transformers" | "allennlp" | "asteroid" | "audiocraft" | "bertopic" | "diffusers" | "doctr" | "espnet" | "fairseq" | "fastai" | "fasttext" | "flair" | "gliner" | "keras" | "k2" | "mindspore" | "ml-agents" | "mlx" | "nemo" | "open_clip" | "paddlenlp" | "peft" | "pyannote-audio" | "pythae" | "sample-factory" | "sentence-transformers" | "setfit" | "spacy" | "span-marker" | "speechbrain" | "stable-baselines3" | "stanza" | "tensorflowtts" | "timm" | "transformers" | "transformers.js" | "unity-sentis" | "whisperkit")[];
+declare const ALL_DISPLAY_MODEL_LIBRARY_KEYS: ("sklearn" | "adapter-transformers" | "allennlp" | "asteroid" | "audiocraft" | "bertopic" | "diffusers" | "doctr" | "espnet" | "fairseq" | "fastai" | "fasttext" | "flair" | "gliner" | "keras" | "k2" | "mindspore" | "ml-agents" | "mlx" | "nemo" | "open_clip" | "paddlenlp" | "peft" | "pyannote-audio" | "pythae" | "sample-factory" | "sentence-transformers" | "setfit" | "spacy" | "span-marker" | "speechbrain" | "stable-baselines3" | "stanza" | "tensorflowtts" | "timm" | "transformers" | "transformers.js" | "unity-sentis" | "whisperkit")[];
 
 /**
  * Mapping from library name (excluding Transformers) to its supported tasks.

package/dist/index.js

@@ -4048,6 +4048,11 @@ var flair = (model) => [
 
 tagger = SequenceTagger.load("${model.id}")`
 ];
+var gliner = (model) => [
+  `from model import GLiNER
+
+model = GLiNER.from_pretrained("${model.id}")`
+];
 var keras = (model) => [
   `from huggingface_hub import from_pretrained_keras
 
@@ -4543,6 +4548,16 @@ var MODEL_LIBRARIES_UI_ELEMENTS = {
       term: { path: "pytorch_model.bin" }
     }
   },
+  gliner: {
+    prettyLabel: "GLiNER",
+    repoName: "GLiNER",
+    repoUrl: "https://github.com/urchade/GLiNER",
+    snippets: gliner,
+    filter: false,
+    countDownloads: {
+      term: { path: "gliner_config.json" }
+    }
+  },
   keras: {
     prettyLabel: "Keras",
     repoName: "Keras",
@@ -4758,6 +4773,14 @@ var MODEL_LIBRARIES_UI_ELEMENTS = {
     countDownloads: {
       wildcard: { path: "*.sentis" }
     }
+  },
+  whisperkit: {
+    prettyLabel: "WhisperKit",
+    repoName: "WhisperKit",
+    repoUrl: "https://github.com/argmaxinc/WhisperKit",
+    countDownloads: {
+      wildcard: { path: "*/model.mil" }
+    }
   }
 };
 var ALL_MODEL_LIBRARY_KEYS = Object.keys(MODEL_LIBRARIES_UI_ELEMENTS);

package/package.json

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

package/src/model-libraries-snippets.ts

@@ -140,6 +140,12 @@ export const flair = (model: ModelData): string[] => [
 tagger = SequenceTagger.load("${model.id}")`,
 ];
 
+export const gliner = (model: ModelData): string[] => [
+	`from model import GLiNER
+
+model = GLiNER.from_pretrained("${model.id}")`,
+];
+
 export const keras = (model: ModelData): string[] => [
 	`from huggingface_hub import from_pretrained_keras
 

package/src/model-libraries.ts

@@ -157,6 +157,16 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = {
 			term: { path: "pytorch_model.bin" },
 		},
 	},
+	gliner: {
+		prettyLabel: "GLiNER",
+		repoName: "GLiNER",
+		repoUrl: "https://github.com/urchade/GLiNER",
+		snippets: snippets.gliner,
+		filter: false,
+		countDownloads: {
+			term: { path: "gliner_config.json" },
+		},
+	},
 	keras: {
 		prettyLabel: "Keras",
 		repoName: "Keras",
@@ -373,6 +383,14 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = {
 			wildcard: { path: "*.sentis" },
 		},
 	},
+	whisperkit: {
+		prettyLabel: "WhisperKit",
+		repoName: "WhisperKit",
+		repoUrl: "https://github.com/argmaxinc/WhisperKit",
+		countDownloads: {
+			wildcard: { path: "*/model.mil" },
+		},
+	},
 } satisfies Record<string, LibraryUiElement>;
 
 export type ModelLibraryKey = keyof typeof MODEL_LIBRARIES_UI_ELEMENTS;