@huggingface/tasks 0.10.16 → 0.10.18

package/dist/src/model-libraries.d.ts

@@ -1,5 +1,5 @@
 import type { ModelData } from "./model-data";
-import type { ElasticBoolQueryFilter } from "./model-libraries-downloads";
+import type { ElasticSearchQuery } from "./model-libraries-downloads";
 /**
  * Elements configurable by a model library.
  */
@@ -32,7 +32,7 @@ export interface LibraryUiElement {
      * By default, those files are counted:
      * "config.json", "config.yaml", "hyperparams.yaml", "meta.yaml"
      */
-    countDownloads?: ElasticBoolQueryFilter;
+    countDownloads?: ElasticSearchQuery;
     /**
      * should we display this library in hf.co/models filter
      * (only for popular libraries with > 100 models)
@@ -61,11 +61,7 @@ export declare const MODEL_LIBRARIES_UI_ELEMENTS: {
         docsUrl: string;
         snippets: (model: ModelData) => string[];
         filter: true;
-        countDownloads: {
-            term: {
-                path: string;
-            };
-        };
+        countDownloads: string;
     };
     allennlp: {
         prettyLabel: string;
@@ -82,11 +78,7 @@ export declare const MODEL_LIBRARIES_UI_ELEMENTS: {
         docsUrl: string;
         snippets: (model: ModelData) => string[];
         filter: true;
-        countDownloads: {
-            term: {
-                path: string;
-            };
-        };
+        countDownloads: string;
     };
     audiocraft: {
         prettyLabel: string;
@@ -95,6 +87,14 @@ export declare const MODEL_LIBRARIES_UI_ELEMENTS: {
         snippets: (model: ModelData) => string[];
         filter: false;
     };
+    audioseal: {
+        prettyLabel: string;
+        repoName: string;
+        repoUrl: string;
+        filter: false;
+        countDownloads: string;
+        snippets: (model: ModelData) => string[];
+    };
     bertopic: {
         prettyLabel: string;
         repoName: string;
@@ -107,11 +107,7 @@ export declare const MODEL_LIBRARIES_UI_ELEMENTS: {
         repoName: string;
         repoUrl: string;
         filter: false;
-        countDownloads: {
-            wildcard: {
-                path: string;
-            };
-        };
+        countDownloads: string;
     };
     bm25s: {
         prettyLabel: string;
@@ -119,23 +115,15 @@ export declare const MODEL_LIBRARIES_UI_ELEMENTS: {
         repoUrl: string;
         snippets: (model: ModelData) => string[];
         filter: false;
-        countDownloads: {
-            term: {
-                path: string;
-            };
-        };
+        countDownloads: string;
     };
     chat_tts: {
         prettyLabel: string;
         repoName: string;
         repoUrl: string;
-        filter: false;
-        countDownloads: {
-            term: {
-                path: string;
-            };
-        };
         snippets: () => string[];
+        filter: false;
+        countDownloads: string;
     };
     diffusers: {
         prettyLabel: string;
@@ -157,22 +145,14 @@ export declare const MODEL_LIBRARIES_UI_ELEMENTS: {
         docsUrl: string;
         filter: false;
         snippets: (model: ModelData) => string[];
-        countDownloads: {
-            wildcard: {
-                path: string;
-            };
-        };
+        countDownloads: string;
     };
     elm: {
         prettyLabel: string;
         repoName: string;
         repoUrl: string;
         filter: false;
-        countDownloads: {
-            wildcard: {
-                path: string;
-            };
-        };
+        countDownloads: string;
     };
     espnet: {
         prettyLabel: string;
@@ -203,11 +183,7 @@ export declare const MODEL_LIBRARIES_UI_ELEMENTS: {
         repoUrl: string;
         snippets: (model: ModelData) => string[];
         filter: true;
-        countDownloads: {
-            wildcard: {
-                path: string;
-            };
-        };
+        countDownloads: string;
     };
     flair: {
         prettyLabel: string;
@@ -216,22 +192,14 @@ export declare const MODEL_LIBRARIES_UI_ELEMENTS: {
         docsUrl: string;
         snippets: (model: ModelData) => string[];
         filter: true;
-        countDownloads: {
-            term: {
-                path: string;
-            };
-        };
+        countDownloads: string;
     };
     "gemma.cpp": {
         prettyLabel: string;
         repoName: string;
         repoUrl: string;
         filter: false;
-        countDownloads: {
-            wildcard: {
-                path: string;
-            };
-        };
+        countDownloads: string;
     };
     gliner: {
         prettyLabel: string;
@@ -239,32 +207,20 @@ export declare const MODEL_LIBRARIES_UI_ELEMENTS: {
         repoUrl: string;
         snippets: (model: ModelData) => string[];
         filter: false;
-        countDownloads: {
-            term: {
-                path: string;
-            };
-        };
+        countDownloads: string;
     };
     grok: {
         prettyLabel: string;
         repoName: string;
         repoUrl: string;
         filter: false;
-        countDownloads: {
-            terms: {
-                path: string[];
-            };
-        };
+        countDownloads: string;
     };
     "hunyuan-dit": {
         prettyLabel: string;
         repoName: string;
         repoUrl: string;
-        countDownloads: {
-            terms: {
-                path: string[];
-            };
-        };
+        countDownloads: string;
     };
     keras: {
         prettyLabel: string;
@@ -273,11 +229,7 @@ export declare const MODEL_LIBRARIES_UI_ELEMENTS: {
         docsUrl: string;
         snippets: (model: ModelData) => string[];
         filter: true;
-        countDownloads: {
-            term: {
-                path: string;
-            };
-        };
+        countDownloads: string;
     };
     "keras-nlp": {
         prettyLabel: string;
@@ -303,11 +255,7 @@ export declare const MODEL_LIBRARIES_UI_ELEMENTS: {
         docsUrl: string;
         snippets: (model: ModelData) => string[];
         filter: true;
-        countDownloads: {
-            wildcard: {
-                path: string;
-            };
-        };
+        countDownloads: string;
     };
     mlx: {
         prettyLabel: string;
@@ -323,11 +271,7 @@ export declare const MODEL_LIBRARIES_UI_ELEMENTS: {
         docsUrl: string;
         snippets: (model: ModelData) => string[];
         filter: false;
-        countDownloads: {
-            term: {
-                path: string;
-            };
-        };
+        countDownloads: string;
     };
     nemo: {
         prettyLabel: string;
@@ -335,11 +279,7 @@ export declare const MODEL_LIBRARIES_UI_ELEMENTS: {
         repoUrl: string;
         snippets: (model: ModelData) => string[];
         filter: true;
-        countDownloads: {
-            wildcard: {
-                path: string;
-            };
-        };
+        countDownloads: string;
     };
     open_clip: {
         prettyLabel: string;
@@ -347,11 +287,7 @@ export declare const MODEL_LIBRARIES_UI_ELEMENTS: {
         repoUrl: string;
         snippets: (model: ModelData) => string[];
         filter: true;
-        countDownloads: {
-            wildcard: {
-                path: string;
-            };
-        };
+        countDownloads: string;
     };
     paddlenlp: {
         prettyLabel: string;
@@ -360,11 +296,7 @@ export declare const MODEL_LIBRARIES_UI_ELEMENTS: {
         docsUrl: string;
         snippets: (model: ModelData) => string[];
         filter: true;
-        countDownloads: {
-            term: {
-                path: string;
-            };
-        };
+        countDownloads: string;
     };
     peft: {
         prettyLabel: string;
@@ -372,11 +304,7 @@ export declare const MODEL_LIBRARIES_UI_ELEMENTS: {
         repoUrl: string;
         snippets: (model: ModelData) => string[];
         filter: true;
-        countDownloads: {
-            term: {
-                path: string;
-            };
-        };
+        countDownloads: string;
     };
     "pyannote-audio": {
         prettyLabel: string;
@@ -397,11 +325,7 @@ export declare const MODEL_LIBRARIES_UI_ELEMENTS: {
         repoName: string;
         repoUrl: string;
         filter: false;
-        countDownloads: {
-            term: {
-                path: string;
-            };
-        };
+        countDownloads: string;
     };
     "sample-factory": {
         prettyLabel: string;
@@ -410,11 +334,7 @@ export declare const MODEL_LIBRARIES_UI_ELEMENTS: {
         docsUrl: string;
         snippets: (model: ModelData) => string[];
         filter: true;
-        countDownloads: {
-            term: {
-                path: string;
-            };
-        };
+        countDownloads: string;
     };
     "sentence-transformers": {
         prettyLabel: string;
@@ -438,11 +358,7 @@ export declare const MODEL_LIBRARIES_UI_ELEMENTS: {
         repoUrl: string;
         snippets: (model: ModelData) => string[];
         filter: true;
-        countDownloads: {
-            term: {
-                path: string;
-            };
-        };
+        countDownloads: string;
     };
     spacy: {
         prettyLabel: string;
@@ -451,11 +367,7 @@ export declare const MODEL_LIBRARIES_UI_ELEMENTS: {
         docsUrl: string;
         snippets: (model: ModelData) => string[];
         filter: true;
-        countDownloads: {
-            wildcard: {
-                path: string;
-            };
-        };
+        countDownloads: string;
     };
     "span-marker": {
         prettyLabel: string;
@@ -472,24 +384,23 @@ export declare const MODEL_LIBRARIES_UI_ELEMENTS: {
         docsUrl: string;
         snippets: (model: ModelData) => string[];
         filter: true;
-        countDownloads: {
-            term: {
-                path: string;
-            };
-        };
+        countDownloads: string;
     };
     "stable-audio-tools": {
         prettyLabel: string;
         repoName: string;
         repoUrl: string;
         filter: false;
-        countDownloads: {
-            term: {
-                path: string;
-            };
-        };
+        countDownloads: string;
         snippets: (model: ModelData) => string[];
     };
+    "diffusion-single-file": {
+        prettyLabel: string;
+        repoName: string;
+        repoUrl: string;
+        filter: false;
+        countDownloads: string;
+    };
     "stable-baselines3": {
         prettyLabel: string;
         repoName: string;
@@ -497,11 +408,7 @@ export declare const MODEL_LIBRARIES_UI_ELEMENTS: {
         docsUrl: string;
         snippets: (model: ModelData) => string[];
         filter: true;
-        countDownloads: {
-            wildcard: {
-                path: string;
-            };
-        };
+        countDownloads: string;
     };
     stanza: {
         prettyLabel: string;
@@ -510,11 +417,7 @@ export declare const MODEL_LIBRARIES_UI_ELEMENTS: {
         docsUrl: string;
         snippets: (model: ModelData) => string[];
         filter: true;
-        countDownloads: {
-            term: {
-                path: string;
-            };
-        };
+        countDownloads: string;
     };
     tensorflowtts: {
         prettyLabel: string;
@@ -527,22 +430,14 @@ export declare const MODEL_LIBRARIES_UI_ELEMENTS: {
         repoName: string;
         repoUrl: string;
         filter: false;
-        countDownloads: {
-            wildcard: {
-                path: string;
-            };
-        };
+        countDownloads: string;
     };
     timesfm: {
         prettyLabel: string;
         repoName: string;
         repoUrl: string;
         filter: false;
-        countDownloads: {
-            term: {
-                path: string;
-            };
-        };
+        countDownloads: string;
     };
     timm: {
         prettyLabel: string;
@@ -551,11 +446,7 @@ export declare const MODEL_LIBRARIES_UI_ELEMENTS: {
         docsUrl: string;
         snippets: (model: ModelData) => string[];
         filter: true;
-        countDownloads: {
-            terms: {
-                path: string[];
-            };
-        };
+        countDownloads: string;
     };
     transformers: {
         prettyLabel: string;
@@ -579,11 +470,7 @@ export declare const MODEL_LIBRARIES_UI_ELEMENTS: {
         repoUrl: string;
         snippets: () => string[];
         filter: true;
-        countDownloads: {
-            wildcard: {
-                path: string;
-            };
-        };
+        countDownloads: string;
     };
     voicecraft: {
         prettyLabel: string;
@@ -596,14 +483,10 @@ export declare const MODEL_LIBRARIES_UI_ELEMENTS: {
         prettyLabel: string;
         repoName: string;
         repoUrl: string;
-        countDownloads: {
-            wildcard: {
-                path: string;
-            };
-        };
+        countDownloads: string;
     };
 };
 export type ModelLibraryKey = keyof typeof MODEL_LIBRARIES_UI_ELEMENTS;
-export declare const ALL_MODEL_LIBRARY_KEYS: ("adapter-transformers" | "allennlp" | "asteroid" | "audiocraft" | "bertopic" | "big_vision" | "bm25s" | "chat_tts" | "diffusers" | "doctr" | "edsnlp" | "elm" | "espnet" | "fairseq" | "fastai" | "fasttext" | "flair" | "gemma.cpp" | "gliner" | "grok" | "hunyuan-dit" | "keras" | "keras-nlp" | "k2" | "mindspore" | "ml-agents" | "mlx" | "mlx-image" | "nemo" | "open_clip" | "paddlenlp" | "peft" | "pyannote-audio" | "pythae" | "recurrentgemma" | "sample-factory" | "sentence-transformers" | "setfit" | "sklearn" | "spacy" | "span-marker" | "speechbrain" | "stable-audio-tools" | "stable-baselines3" | "stanza" | "tensorflowtts" | "tic-clip" | "timesfm" | "timm" | "transformers" | "transformers.js" | "unity-sentis" | "voicecraft" | "whisperkit")[];
-export declare const ALL_DISPLAY_MODEL_LIBRARY_KEYS: ("adapter-transformers" | "allennlp" | "asteroid" | "audiocraft" | "bertopic" | "big_vision" | "bm25s" | "chat_tts" | "diffusers" | "doctr" | "edsnlp" | "elm" | "espnet" | "fairseq" | "fastai" | "fasttext" | "flair" | "gemma.cpp" | "gliner" | "grok" | "hunyuan-dit" | "keras" | "keras-nlp" | "k2" | "mindspore" | "ml-agents" | "mlx" | "mlx-image" | "nemo" | "open_clip" | "paddlenlp" | "peft" | "pyannote-audio" | "pythae" | "recurrentgemma" | "sample-factory" | "sentence-transformers" | "setfit" | "sklearn" | "spacy" | "span-marker" | "speechbrain" | "stable-audio-tools" | "stable-baselines3" | "stanza" | "tensorflowtts" | "tic-clip" | "timesfm" | "timm" | "transformers" | "transformers.js" | "unity-sentis" | "voicecraft" | "whisperkit")[];
+export declare const ALL_MODEL_LIBRARY_KEYS: ("adapter-transformers" | "allennlp" | "asteroid" | "audiocraft" | "audioseal" | "bertopic" | "big_vision" | "bm25s" | "chat_tts" | "diffusers" | "doctr" | "edsnlp" | "elm" | "espnet" | "fairseq" | "fastai" | "fasttext" | "flair" | "gemma.cpp" | "gliner" | "grok" | "hunyuan-dit" | "keras" | "keras-nlp" | "k2" | "mindspore" | "ml-agents" | "mlx" | "mlx-image" | "nemo" | "open_clip" | "paddlenlp" | "peft" | "pyannote-audio" | "pythae" | "recurrentgemma" | "sample-factory" | "sentence-transformers" | "setfit" | "sklearn" | "spacy" | "span-marker" | "speechbrain" | "stable-audio-tools" | "diffusion-single-file" | "stable-baselines3" | "stanza" | "tensorflowtts" | "tic-clip" | "timesfm" | "timm" | "transformers" | "transformers.js" | "unity-sentis" | "voicecraft" | "whisperkit")[];
+export declare const ALL_DISPLAY_MODEL_LIBRARY_KEYS: ("adapter-transformers" | "allennlp" | "asteroid" | "audiocraft" | "audioseal" | "bertopic" | "big_vision" | "bm25s" | "chat_tts" | "diffusers" | "doctr" | "edsnlp" | "elm" | "espnet" | "fairseq" | "fastai" | "fasttext" | "flair" | "gemma.cpp" | "gliner" | "grok" | "hunyuan-dit" | "keras" | "keras-nlp" | "k2" | "mindspore" | "ml-agents" | "mlx" | "mlx-image" | "nemo" | "open_clip" | "paddlenlp" | "peft" | "pyannote-audio" | "pythae" | "recurrentgemma" | "sample-factory" | "sentence-transformers" | "setfit" | "sklearn" | "spacy" | "span-marker" | "speechbrain" | "stable-audio-tools" | "diffusion-single-file" | "stable-baselines3" | "stanza" | "tensorflowtts" | "tic-clip" | "timesfm" | "timm" | "transformers" | "transformers.js" | "unity-sentis" | "voicecraft" | "whisperkit")[];
 //# sourceMappingURL=model-libraries.d.ts.map

package/dist/src/model-libraries.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"model-libraries.d.ts","sourceRoot":"","sources":["../../src/model-libraries.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,cAAc,CAAC;AAC9C,OAAO,KAAK,EAAE,sBAAsB,EAAE,MAAM,6BAA6B,CAAC;AAE1E;;GAEG;AACH,MAAM,WAAW,gBAAgB;IAChC;;;;OAIG;IACH,WAAW,EAAE,MAAM,CAAC;IACpB;;OAEG;IACH,QAAQ,EAAE,MAAM,CAAC;IACjB;;OAEG;IACH,OAAO,EAAE,MAAM,CAAC;IAChB;;OAEG;IACH,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB;;OAEG;IACH,QAAQ,CAAC,EAAE,CAAC,KAAK,EAAE,SAAS,KAAK,MAAM,EAAE,CAAC;IAC1C;;;;;OAKG;IACH,cAAc,CAAC,EAAE,sBAAsB,CAAC;IACxC;;;OAGG;IACH,MAAM,CAAC,EAAE,OAAO,CAAC;CACjB;AAED;;;;;;;;;;;;;GAaG;AAEH,eAAO,MAAM,2BAA2B;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CA8cI,CAAC;AAE7C,MAAM,MAAM,eAAe,GAAG,MAAM,OAAO,2BAA2B,CAAC;AAEvE,eAAO,MAAM,sBAAsB,4uBAAgE,CAAC;AAEpG,eAAO,MAAM,8BAA8B,4uBAQ1B,CAAC"}
+{"version":3,"file":"model-libraries.d.ts","sourceRoot":"","sources":["../../src/model-libraries.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,cAAc,CAAC;AAC9C,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,6BAA6B,CAAC;AAEtE;;GAEG;AACH,MAAM,WAAW,gBAAgB;IAChC;;;;OAIG;IACH,WAAW,EAAE,MAAM,CAAC;IACpB;;OAEG;IACH,QAAQ,EAAE,MAAM,CAAC;IACjB;;OAEG;IACH,OAAO,EAAE,MAAM,CAAC;IAChB;;OAEG;IACH,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB;;OAEG;IACH,QAAQ,CAAC,EAAE,CAAC,KAAK,EAAE,SAAS,KAAK,MAAM,EAAE,CAAC;IAC1C;;;;;OAKG;IACH,cAAc,CAAC,EAAE,kBAAkB,CAAC;IACpC;;;OAGG;IACH,MAAM,CAAC,EAAE,OAAO,CAAC;CACjB;AAED;;;;;;;;;;;;;GAaG;AAEH,eAAO,MAAM,2BAA2B;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAibI,CAAC;AAE7C,MAAM,MAAM,eAAe,GAAG,MAAM,OAAO,2BAA2B,CAAC;AAEvE,eAAO,MAAM,sBAAsB,oxBAAgE,CAAC;AAEpG,eAAO,MAAM,8BAA8B,oxBAQ1B,CAAC"}

package/package.json

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

package/src/model-libraries-downloads.ts

@@ -6,15 +6,13 @@
  * Read this doc about download stats on the Hub:
  *
  * https://huggingface.co/docs/hub/models-download-stats
- *
+ * Available fields:
+ *  - path: the complete file path (relative) (e.g: "prefix/file.extension")
+ *  - path_prefix: the prefix of the file path (e.g: "prefix/", empty if no prefix)
+ *  - path_extension: the extension of the file path (e.g: "extension")
+ *  - path_filename: the name of the file path (e.g: "file")
  * see also:
- * https://www.elastic.co/guide/en/elasticsearch/reference/current/query-dsl-bool-query.html
+ * https://www.elastic.co/guide/en/elasticsearch/reference/current/query-dsl-query-string-query.html
  */
 
-export type ElasticBoolQueryFilter =
-	// match a single filename
-	| { term?: { path: string } }
-	// match multiple possible filenames
-	| { terms?: { path: string[] } }
-	// match a wildcard
-	| { wildcard?: { path: string } };
+export type ElasticSearchQuery = string;

package/src/model-libraries-snippets.ts

@@ -46,6 +46,26 @@ export const asteroid = (model: ModelData): string[] => [
 model = BaseModel.from_pretrained("${model.id}")`,
 ];
 
+export const audioseal = (model: ModelData): string[] => {
+	const watermarkSnippet = `# Watermark Generator
+from audioseal import AudioSeal
+
+model = AudioSeal.load_generator("${model.id}")
+# pass a tensor (tensor_wav) of shape (batch, channels, samples) and a sample rate
+wav, sr = tensor_wav, 16000
+	
+watermark = model.get_watermark(wav, sr)
+watermarked_audio = wav + watermark`;
+
+	const detectorSnippet = `# Watermark Detector
+from audioseal import AudioSeal
+
+detector = AudioSeal.load_detector("${model.id}")
+	
+result, message = detector.detect_watermark(watermarked_audio, sr)`;
+	return [watermarkSnippet, detectorSnippet];
+};
+
 function get_base_diffusers_model(model: ModelData): string {
 	return model.cardData?.base_model?.toString() ?? "fill-in-base-model";
 }