@huggingface/tasks 0.10.14 → 0.10.15

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
package/dist/index.cjs CHANGED
@@ -4125,6 +4125,23 @@ var diffusers = (model) => {
4125
4125
  return diffusers_default(model);
4126
4126
  }
4127
4127
  };
4128
+ var edsnlp = (model) => {
4129
+ const packageName = nameWithoutNamespace(model.id).replaceAll("-", "_");
4130
+ return [
4131
+ `# Load it from the Hub directly
4132
+ import edsnlp
4133
+ nlp = edsnlp.load("${model.id}")
4134
+ `,
4135
+ `# Or install it as a package
4136
+ !pip install git+https://huggingface.co/${model.id}
4137
+
4138
+ # and import it as a module
4139
+ import ${packageName}
4140
+
4141
+ nlp = ${packageName}.load() # or edsnlp.load("${packageName}")
4142
+ `
4143
+ ];
4144
+ };
4128
4145
  var espnetTTS = (model) => [
4129
4146
  `from espnet2.bin.tts_inference import Text2Speech
4130
4147
 
@@ -4710,6 +4727,17 @@ var MODEL_LIBRARIES_UI_ELEMENTS = {
4710
4727
  repoName: "doctr",
4711
4728
  repoUrl: "https://github.com/mindee/doctr"
4712
4729
  },
4730
+ edsnlp: {
4731
+ prettyLabel: "EDS-NLP",
4732
+ repoName: "edsnlp",
4733
+ repoUrl: "https://github.com/aphp/edsnlp",
4734
+ docsUrl: "https://aphp.github.io/edsnlp/latest/",
4735
+ filter: false,
4736
+ snippets: edsnlp,
4737
+ countDownloads: {
4738
+ wildcard: { path: "*/config.cfg" }
4739
+ }
4740
+ },
4713
4741
  elm: {
4714
4742
  prettyLabel: "ELM",
4715
4743
  repoName: "elm",
@@ -4763,6 +4791,13 @@ var MODEL_LIBRARIES_UI_ELEMENTS = {
4763
4791
  term: { path: "pytorch_model.bin" }
4764
4792
  }
4765
4793
  },
4794
+ "gemma.cpp": {
4795
+ prettyLabel: "gemma.cpp",
4796
+ repoName: "gemma.cpp",
4797
+ repoUrl: "https://github.com/google/gemma.cpp",
4798
+ filter: false,
4799
+ countDownloads: { wildcard: { path: "*.sbs" } }
4800
+ },
4766
4801
  gliner: {
4767
4802
  prettyLabel: "GLiNER",
4768
4803
  repoName: "GLiNER",
@@ -4884,6 +4919,13 @@ var MODEL_LIBRARIES_UI_ELEMENTS = {
4884
4919
  snippets: pythae,
4885
4920
  filter: true
4886
4921
  },
4922
+ recurrentgemma: {
4923
+ prettyLabel: "RecurrentGemma",
4924
+ repoName: "recurrentgemma",
4925
+ repoUrl: "https://github.com/google-deepmind/recurrentgemma",
4926
+ filter: false,
4927
+ countDownloads: { term: { path: "tokenizer.model" } }
4928
+ },
4887
4929
  "sample-factory": {
4888
4930
  prettyLabel: "sample-factory",
4889
4931
  repoName: "sample-factory",
@@ -4985,6 +5027,13 @@ var MODEL_LIBRARIES_UI_ELEMENTS = {
4985
5027
  repoUrl: "https://github.com/TensorSpeech/TensorFlowTTS",
4986
5028
  snippets: tensorflowtts
4987
5029
  },
5030
+ "tic-clip": {
5031
+ prettyLabel: "TiC-CLIP",
5032
+ repoName: "TiC-CLIP",
5033
+ repoUrl: "https://github.com/apple/ml-tic-clip",
5034
+ filter: false,
5035
+ countDownloads: { wildcard: { path: "checkpoints/*.pt" } }
5036
+ },
4988
5037
  timesfm: {
4989
5038
  prettyLabel: "TimesFM",
4990
5039
  repoName: "timesfm",
@@ -6047,7 +6096,7 @@ var LOCAL_APPS = {
6047
6096
  // src/dataset-libraries.ts
6048
6097
  var DATASET_LIBRARIES_UI_ELEMENTS = {
6049
6098
  mlcroissant: {
6050
- prettyLabel: "mlcroissant",
6099
+ prettyLabel: "Croissant",
6051
6100
  repoName: "croissant",
6052
6101
  repoUrl: "https://github.com/mlcommons/croissant/tree/main/python/mlcroissant",
6053
6102
  docsUrl: "https://github.com/mlcommons/croissant/blob/main/python/mlcroissant/README.md"
package/dist/index.js CHANGED
@@ -4086,6 +4086,23 @@ var diffusers = (model) => {
4086
4086
  return diffusers_default(model);
4087
4087
  }
4088
4088
  };
4089
+ var edsnlp = (model) => {
4090
+ const packageName = nameWithoutNamespace(model.id).replaceAll("-", "_");
4091
+ return [
4092
+ `# Load it from the Hub directly
4093
+ import edsnlp
4094
+ nlp = edsnlp.load("${model.id}")
4095
+ `,
4096
+ `# Or install it as a package
4097
+ !pip install git+https://huggingface.co/${model.id}
4098
+
4099
+ # and import it as a module
4100
+ import ${packageName}
4101
+
4102
+ nlp = ${packageName}.load() # or edsnlp.load("${packageName}")
4103
+ `
4104
+ ];
4105
+ };
4089
4106
  var espnetTTS = (model) => [
4090
4107
  `from espnet2.bin.tts_inference import Text2Speech
4091
4108
 
@@ -4671,6 +4688,17 @@ var MODEL_LIBRARIES_UI_ELEMENTS = {
4671
4688
  repoName: "doctr",
4672
4689
  repoUrl: "https://github.com/mindee/doctr"
4673
4690
  },
4691
+ edsnlp: {
4692
+ prettyLabel: "EDS-NLP",
4693
+ repoName: "edsnlp",
4694
+ repoUrl: "https://github.com/aphp/edsnlp",
4695
+ docsUrl: "https://aphp.github.io/edsnlp/latest/",
4696
+ filter: false,
4697
+ snippets: edsnlp,
4698
+ countDownloads: {
4699
+ wildcard: { path: "*/config.cfg" }
4700
+ }
4701
+ },
4674
4702
  elm: {
4675
4703
  prettyLabel: "ELM",
4676
4704
  repoName: "elm",
@@ -4724,6 +4752,13 @@ var MODEL_LIBRARIES_UI_ELEMENTS = {
4724
4752
  term: { path: "pytorch_model.bin" }
4725
4753
  }
4726
4754
  },
4755
+ "gemma.cpp": {
4756
+ prettyLabel: "gemma.cpp",
4757
+ repoName: "gemma.cpp",
4758
+ repoUrl: "https://github.com/google/gemma.cpp",
4759
+ filter: false,
4760
+ countDownloads: { wildcard: { path: "*.sbs" } }
4761
+ },
4727
4762
  gliner: {
4728
4763
  prettyLabel: "GLiNER",
4729
4764
  repoName: "GLiNER",
@@ -4845,6 +4880,13 @@ var MODEL_LIBRARIES_UI_ELEMENTS = {
4845
4880
  snippets: pythae,
4846
4881
  filter: true
4847
4882
  },
4883
+ recurrentgemma: {
4884
+ prettyLabel: "RecurrentGemma",
4885
+ repoName: "recurrentgemma",
4886
+ repoUrl: "https://github.com/google-deepmind/recurrentgemma",
4887
+ filter: false,
4888
+ countDownloads: { term: { path: "tokenizer.model" } }
4889
+ },
4848
4890
  "sample-factory": {
4849
4891
  prettyLabel: "sample-factory",
4850
4892
  repoName: "sample-factory",
@@ -4946,6 +4988,13 @@ var MODEL_LIBRARIES_UI_ELEMENTS = {
4946
4988
  repoUrl: "https://github.com/TensorSpeech/TensorFlowTTS",
4947
4989
  snippets: tensorflowtts
4948
4990
  },
4991
+ "tic-clip": {
4992
+ prettyLabel: "TiC-CLIP",
4993
+ repoName: "TiC-CLIP",
4994
+ repoUrl: "https://github.com/apple/ml-tic-clip",
4995
+ filter: false,
4996
+ countDownloads: { wildcard: { path: "checkpoints/*.pt" } }
4997
+ },
4949
4998
  timesfm: {
4950
4999
  prettyLabel: "TimesFM",
4951
5000
  repoName: "timesfm",
@@ -6008,7 +6057,7 @@ var LOCAL_APPS = {
6008
6057
  // src/dataset-libraries.ts
6009
6058
  var DATASET_LIBRARIES_UI_ELEMENTS = {
6010
6059
  mlcroissant: {
6011
- prettyLabel: "mlcroissant",
6060
+ prettyLabel: "Croissant",
6012
6061
  repoName: "croissant",
6013
6062
  repoUrl: "https://github.com/mlcommons/croissant/tree/main/python/mlcroissant",
6014
6063
  docsUrl: "https://github.com/mlcommons/croissant/blob/main/python/mlcroissant/README.md"
@@ -4,6 +4,7 @@ export declare const allennlp: (model: ModelData) => string[];
4
4
  export declare const asteroid: (model: ModelData) => string[];
5
5
  export declare const bertopic: (model: ModelData) => string[];
6
6
  export declare const diffusers: (model: ModelData) => string[];
7
+ export declare const edsnlp: (model: ModelData) => string[];
7
8
  export declare const espnetTTS: (model: ModelData) => string[];
8
9
  export declare const espnetASR: (model: ModelData) => string[];
9
10
  export declare const espnet: (model: ModelData) => string[];
@@ -1 +1 @@
1
- {"version":3,"file":"model-libraries-snippets.d.ts","sourceRoot":"","sources":["../../src/model-libraries-snippets.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,cAAc,CAAC;AAY9C,eAAO,MAAM,QAAQ,UAAW,SAAS,KAAG,MAAM,EAKjD,CAAC;AAkBF,eAAO,MAAM,QAAQ,UAAW,SAAS,KAAG,MAAM,EAKjD,CAAC;AAEF,eAAO,MAAM,QAAQ,UAAW,SAAS,KAAG,MAAM,EAIjD,CAAC;AAMF,eAAO,MAAM,QAAQ,UAAW,SAAS,KAAG,MAAM,EAIjD,CAAC;AA+BF,eAAO,MAAM,SAAS,UAAW,SAAS,KAAG,MAAM,EAUlD,CAAC;AAEF,eAAO,MAAM,SAAS,UAAW,SAAS,KAAG,MAAM,EAMlD,CAAC;AAEF,eAAO,MAAM,SAAS,UAAW,SAAS,KAAG,MAAM,EASlD,CAAC;AAIF,eAAO,MAAM,MAAM,UAAW,SAAS,KAAG,MAAM,EAO/C,CAAC;AAEF,eAAO,MAAM,OAAO,UAAW,SAAS,KAAG,MAAM,EAMhD,CAAC;AAEF,eAAO,MAAM,KAAK,UAAW,SAAS,KAAG,MAAM,EAI9C,CAAC;AAEF,eAAO,MAAM,MAAM,UAAW,SAAS,KAAG,MAAM,EAI/C,CAAC;AAEF,eAAO,MAAM,KAAK,UAAW,SAAS,KAAG,MAAM,EAK9C,CAAC;AAEF,eAAO,MAAM,SAAS,UAAW,SAAS,KAAG,MAAM,EASlD,CAAC;AAEF,eAAO,MAAM,SAAS,UAAW,SAAS,KAAG,MAAM,EAKlD,CAAC;AAEF,eAAO,MAAM,SAAS,UAAW,SAAS,KAAG,MAAM,EAsBlD,CAAC;AAEF,eAAO,MAAM,uBAAuB,UAAW,SAAS,KAAG,MAAM,EAehE,CAAC;AAiBF,eAAO,MAAM,cAAc,UAAW,SAAS,KAAG,MAAM,EAKvD,CAAC;AAyBF,eAAO,MAAM,aAAa,UAAW,SAAS,KAAG,MAAM,EAOtD,CAAC;AAEF,eAAO,MAAM,IAAI,UAAW,SAAS,KAAG,MAAM,EAI7C,CAAC;AAsCF,eAAO,MAAM,OAAO,UAAW,SAAS,KAAG,MAAM,EAehD,CAAC;AAEF,eAAO,MAAM,kBAAkB,UAAW,SAAS,KAAG,MAAM,EAmC3D,CAAC;AAEF,eAAO,MAAM,MAAM,UAAW,SAAS,KAAG,MAAM,EAI/C,CAAC;AAEF,eAAO,MAAM,aAAa,UAAW,SAAS,KAAG,MAAM,EAEtD,CAAC;AAEF,eAAO,MAAM,oBAAoB,UAAW,SAAS,KAAG,MAAM,EAI7D,CAAC;AAEF,eAAO,MAAM,MAAM,UAAW,SAAS,KAAG,MAAM,EAI/C,CAAC;AAEF,eAAO,MAAM,KAAK,UAAW,SAAS,KAAG,MAAM,EAU9C,CAAC;AAEF,eAAO,MAAM,WAAW,UAAW,SAAS,KAAG,MAAM,EAIpD,CAAC;AAEF,eAAO,MAAM,MAAM,UAAW,SAAS,KAAG,MAAM,EAK/C,CAAC;AAkBF,eAAO,MAAM,WAAW,UAAW,SAAS,KAAG,MAAM,EAkBpD,CAAC;AAEF,eAAO,MAAM,YAAY,UAAW,SAAS,KAAG,MAAM,EA4CrD,CAAC;AAEF,eAAO,MAAM,cAAc,UAAW,SAAS,KAAG,MAAM,EAcvD,CAAC;AAiBF,eAAO,MAAM,IAAI,UAAW,SAAS,KAAG,MAAM,EAkB7C,CAAC;AAEF,eAAO,MAAM,QAAQ,UAAW,SAAS,KAAG,MAAM,EAKjD,CAAC;AAEF,eAAO,MAAM,gBAAgB,UAAW,SAAS,KAAG,MAAM,EAMzD,CAAC;AAgBF,eAAO,MAAM,QAAQ,UAAW,SAAS,KAAG,MAAM,EAEjD,CAAC;AAEF,eAAO,MAAM,MAAM,QAA6B,MAAM,EAMrD,CAAC;AAEF,eAAO,MAAM,UAAU,UAAW,SAAS,KAAG,MAAM,EAInD,CAAC;AAEF,eAAO,MAAM,OAAO,QAAO,MAAM,EAYhC,CAAC;AAEF,eAAO,MAAM,GAAG,UAAW,SAAS,KAAG,MAAM,EAK5C,CAAC;AAEF,eAAO,MAAM,KAAK,UAAW,SAAS,KAAG,MAAM,EAI9C,CAAC;AAEF,eAAO,MAAM,IAAI,UAAW,SAAS,KAAG,MAAM,EAQ7C,CAAC;AAEF,eAAO,MAAM,MAAM,UAAW,SAAS,KAAG,MAAM,EAI/C,CAAC;AA6BF,eAAO,MAAM,UAAU,UAAW,SAAS,KAAG,MAAM,EAUnD,CAAC"}
1
+ {"version":3,"file":"model-libraries-snippets.d.ts","sourceRoot":"","sources":["../../src/model-libraries-snippets.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,cAAc,CAAC;AAY9C,eAAO,MAAM,QAAQ,UAAW,SAAS,KAAG,MAAM,EAKjD,CAAC;AAkBF,eAAO,MAAM,QAAQ,UAAW,SAAS,KAAG,MAAM,EAKjD,CAAC;AAEF,eAAO,MAAM,QAAQ,UAAW,SAAS,KAAG,MAAM,EAIjD,CAAC;AAMF,eAAO,MAAM,QAAQ,UAAW,SAAS,KAAG,MAAM,EAIjD,CAAC;AA+BF,eAAO,MAAM,SAAS,UAAW,SAAS,KAAG,MAAM,EAUlD,CAAC;AAEF,eAAO,MAAM,MAAM,UAAW,SAAS,KAAG,MAAM,EAgB/C,CAAC;AAEF,eAAO,MAAM,SAAS,UAAW,SAAS,KAAG,MAAM,EAMlD,CAAC;AAEF,eAAO,MAAM,SAAS,UAAW,SAAS,KAAG,MAAM,EASlD,CAAC;AAIF,eAAO,MAAM,MAAM,UAAW,SAAS,KAAG,MAAM,EAO/C,CAAC;AAEF,eAAO,MAAM,OAAO,UAAW,SAAS,KAAG,MAAM,EAMhD,CAAC;AAEF,eAAO,MAAM,KAAK,UAAW,SAAS,KAAG,MAAM,EAI9C,CAAC;AAEF,eAAO,MAAM,MAAM,UAAW,SAAS,KAAG,MAAM,EAI/C,CAAC;AAEF,eAAO,MAAM,KAAK,UAAW,SAAS,KAAG,MAAM,EAK9C,CAAC;AAEF,eAAO,MAAM,SAAS,UAAW,SAAS,KAAG,MAAM,EASlD,CAAC;AAEF,eAAO,MAAM,SAAS,UAAW,SAAS,KAAG,MAAM,EAKlD,CAAC;AAEF,eAAO,MAAM,SAAS,UAAW,SAAS,KAAG,MAAM,EAsBlD,CAAC;AAEF,eAAO,MAAM,uBAAuB,UAAW,SAAS,KAAG,MAAM,EAehE,CAAC;AAiBF,eAAO,MAAM,cAAc,UAAW,SAAS,KAAG,MAAM,EAKvD,CAAC;AAyBF,eAAO,MAAM,aAAa,UAAW,SAAS,KAAG,MAAM,EAOtD,CAAC;AAEF,eAAO,MAAM,IAAI,UAAW,SAAS,KAAG,MAAM,EAI7C,CAAC;AAsCF,eAAO,MAAM,OAAO,UAAW,SAAS,KAAG,MAAM,EAehD,CAAC;AAEF,eAAO,MAAM,kBAAkB,UAAW,SAAS,KAAG,MAAM,EAmC3D,CAAC;AAEF,eAAO,MAAM,MAAM,UAAW,SAAS,KAAG,MAAM,EAI/C,CAAC;AAEF,eAAO,MAAM,aAAa,UAAW,SAAS,KAAG,MAAM,EAEtD,CAAC;AAEF,eAAO,MAAM,oBAAoB,UAAW,SAAS,KAAG,MAAM,EAI7D,CAAC;AAEF,eAAO,MAAM,MAAM,UAAW,SAAS,KAAG,MAAM,EAI/C,CAAC;AAEF,eAAO,MAAM,KAAK,UAAW,SAAS,KAAG,MAAM,EAU9C,CAAC;AAEF,eAAO,MAAM,WAAW,UAAW,SAAS,KAAG,MAAM,EAIpD,CAAC;AAEF,eAAO,MAAM,MAAM,UAAW,SAAS,KAAG,MAAM,EAK/C,CAAC;AAkBF,eAAO,MAAM,WAAW,UAAW,SAAS,KAAG,MAAM,EAkBpD,CAAC;AAEF,eAAO,MAAM,YAAY,UAAW,SAAS,KAAG,MAAM,EA4CrD,CAAC;AAEF,eAAO,MAAM,cAAc,UAAW,SAAS,KAAG,MAAM,EAcvD,CAAC;AAiBF,eAAO,MAAM,IAAI,UAAW,SAAS,KAAG,MAAM,EAkB7C,CAAC;AAEF,eAAO,MAAM,QAAQ,UAAW,SAAS,KAAG,MAAM,EAKjD,CAAC;AAEF,eAAO,MAAM,gBAAgB,UAAW,SAAS,KAAG,MAAM,EAMzD,CAAC;AAgBF,eAAO,MAAM,QAAQ,UAAW,SAAS,KAAG,MAAM,EAEjD,CAAC;AAEF,eAAO,MAAM,MAAM,QAA6B,MAAM,EAMrD,CAAC;AAEF,eAAO,MAAM,UAAU,UAAW,SAAS,KAAG,MAAM,EAInD,CAAC;AAEF,eAAO,MAAM,OAAO,QAAO,MAAM,EAYhC,CAAC;AAEF,eAAO,MAAM,GAAG,UAAW,SAAS,KAAG,MAAM,EAK5C,CAAC;AAEF,eAAO,MAAM,KAAK,UAAW,SAAS,KAAG,MAAM,EAI9C,CAAC;AAEF,eAAO,MAAM,IAAI,UAAW,SAAS,KAAG,MAAM,EAQ7C,CAAC;AAEF,eAAO,MAAM,MAAM,UAAW,SAAS,KAAG,MAAM,EAI/C,CAAC;AA6BF,eAAO,MAAM,UAAU,UAAW,SAAS,KAAG,MAAM,EAUnD,CAAC"}
@@ -138,6 +138,19 @@ export declare const MODEL_LIBRARIES_UI_ELEMENTS: {
138
138
  repoName: string;
139
139
  repoUrl: string;
140
140
  };
141
+ edsnlp: {
142
+ prettyLabel: string;
143
+ repoName: string;
144
+ repoUrl: string;
145
+ docsUrl: string;
146
+ filter: false;
147
+ snippets: (model: ModelData) => string[];
148
+ countDownloads: {
149
+ wildcard: {
150
+ path: string;
151
+ };
152
+ };
153
+ };
141
154
  elm: {
142
155
  prettyLabel: string;
143
156
  repoName: string;
@@ -197,6 +210,17 @@ export declare const MODEL_LIBRARIES_UI_ELEMENTS: {
197
210
  };
198
211
  };
199
212
  };
213
+ "gemma.cpp": {
214
+ prettyLabel: string;
215
+ repoName: string;
216
+ repoUrl: string;
217
+ filter: false;
218
+ countDownloads: {
219
+ wildcard: {
220
+ path: string;
221
+ };
222
+ };
223
+ };
200
224
  gliner: {
201
225
  prettyLabel: string;
202
226
  repoName: string;
@@ -346,6 +370,17 @@ export declare const MODEL_LIBRARIES_UI_ELEMENTS: {
346
370
  snippets: (model: ModelData) => string[];
347
371
  filter: true;
348
372
  };
373
+ recurrentgemma: {
374
+ prettyLabel: string;
375
+ repoName: string;
376
+ repoUrl: string;
377
+ filter: false;
378
+ countDownloads: {
379
+ term: {
380
+ path: string;
381
+ };
382
+ };
383
+ };
349
384
  "sample-factory": {
350
385
  prettyLabel: string;
351
386
  repoName: string;
@@ -465,6 +500,17 @@ export declare const MODEL_LIBRARIES_UI_ELEMENTS: {
465
500
  repoUrl: string;
466
501
  snippets: (model: ModelData) => string[];
467
502
  };
503
+ "tic-clip": {
504
+ prettyLabel: string;
505
+ repoName: string;
506
+ repoUrl: string;
507
+ filter: false;
508
+ countDownloads: {
509
+ wildcard: {
510
+ path: string;
511
+ };
512
+ };
513
+ };
468
514
  timesfm: {
469
515
  prettyLabel: string;
470
516
  repoName: string;
@@ -536,6 +582,6 @@ export declare const MODEL_LIBRARIES_UI_ELEMENTS: {
536
582
  };
537
583
  };
538
584
  export type ModelLibraryKey = keyof typeof MODEL_LIBRARIES_UI_ELEMENTS;
539
- export declare const ALL_MODEL_LIBRARY_KEYS: ("adapter-transformers" | "allennlp" | "asteroid" | "audiocraft" | "bertopic" | "big_vision" | "chat_tts" | "diffusers" | "doctr" | "elm" | "espnet" | "fairseq" | "fastai" | "fasttext" | "flair" | "gliner" | "grok" | "keras" | "keras-nlp" | "k2" | "mindspore" | "ml-agents" | "mlx" | "mlx-image" | "nemo" | "open_clip" | "paddlenlp" | "peft" | "pyannote-audio" | "pythae" | "sample-factory" | "sentence-transformers" | "setfit" | "sklearn" | "spacy" | "span-marker" | "speechbrain" | "stable-audio-tools" | "stable-baselines3" | "stanza" | "tensorflowtts" | "timesfm" | "timm" | "transformers" | "transformers.js" | "unity-sentis" | "voicecraft" | "whisperkit")[];
540
- export declare const ALL_DISPLAY_MODEL_LIBRARY_KEYS: ("adapter-transformers" | "allennlp" | "asteroid" | "audiocraft" | "bertopic" | "big_vision" | "chat_tts" | "diffusers" | "doctr" | "elm" | "espnet" | "fairseq" | "fastai" | "fasttext" | "flair" | "gliner" | "grok" | "keras" | "keras-nlp" | "k2" | "mindspore" | "ml-agents" | "mlx" | "mlx-image" | "nemo" | "open_clip" | "paddlenlp" | "peft" | "pyannote-audio" | "pythae" | "sample-factory" | "sentence-transformers" | "setfit" | "sklearn" | "spacy" | "span-marker" | "speechbrain" | "stable-audio-tools" | "stable-baselines3" | "stanza" | "tensorflowtts" | "timesfm" | "timm" | "transformers" | "transformers.js" | "unity-sentis" | "voicecraft" | "whisperkit")[];
585
+ export declare const ALL_MODEL_LIBRARY_KEYS: ("adapter-transformers" | "allennlp" | "asteroid" | "audiocraft" | "bertopic" | "big_vision" | "chat_tts" | "diffusers" | "doctr" | "edsnlp" | "elm" | "espnet" | "fairseq" | "fastai" | "fasttext" | "flair" | "gemma.cpp" | "gliner" | "grok" | "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")[];
586
+ export declare const ALL_DISPLAY_MODEL_LIBRARY_KEYS: ("adapter-transformers" | "allennlp" | "asteroid" | "audiocraft" | "bertopic" | "big_vision" | "chat_tts" | "diffusers" | "doctr" | "edsnlp" | "elm" | "espnet" | "fairseq" | "fastai" | "fasttext" | "flair" | "gemma.cpp" | "gliner" | "grok" | "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")[];
541
587
  //# sourceMappingURL=model-libraries.d.ts.map
@@ -1 +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;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CA4ZI,CAAC;AAE7C,MAAM,MAAM,eAAe,GAAG,MAAM,OAAO,2BAA2B,CAAC;AAEvE,eAAO,MAAM,sBAAsB,ypBAAgE,CAAC;AAEpG,eAAO,MAAM,8BAA8B,ypBAQ1B,CAAC"}
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;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CA4bI,CAAC;AAE7C,MAAM,MAAM,eAAe,GAAG,MAAM,OAAO,2BAA2B,CAAC;AAEvE,eAAO,MAAM,sBAAsB,ktBAAgE,CAAC;AAEpG,eAAO,MAAM,8BAA8B,ktBAQ1B,CAAC"}
package/package.json CHANGED
@@ -1,7 +1,7 @@
1
1
  {
2
2
  "name": "@huggingface/tasks",
3
3
  "packageManager": "pnpm@8.10.5",
4
- "version": "0.10.14",
4
+ "version": "0.10.15",
5
5
  "description": "List of ML tasks for huggingface.co/tasks",
6
6
  "repository": "https://github.com/huggingface/huggingface.js.git",
7
7
  "publishConfig": {
@@ -24,7 +24,7 @@ export interface DatasetLibraryUiElement {
24
24
 
25
25
  export const DATASET_LIBRARIES_UI_ELEMENTS = {
26
26
  mlcroissant: {
27
- prettyLabel: "mlcroissant",
27
+ prettyLabel: "Croissant",
28
28
  repoName: "croissant",
29
29
  repoUrl: "https://github.com/mlcommons/croissant/tree/main/python/mlcroissant",
30
30
  docsUrl: "https://github.com/mlcommons/croissant/blob/main/python/mlcroissant/README.md",
@@ -97,6 +97,24 @@ export const diffusers = (model: ModelData): string[] => {
97
97
  }
98
98
  };
99
99
 
100
+ export const edsnlp = (model: ModelData): string[] => {
101
+ const packageName = nameWithoutNamespace(model.id).replaceAll("-", "_");
102
+ return [
103
+ `# Load it from the Hub directly
104
+ import edsnlp
105
+ nlp = edsnlp.load("${model.id}")
106
+ `,
107
+ `# Or install it as a package
108
+ !pip install git+https://huggingface.co/${model.id}
109
+
110
+ # and import it as a module
111
+ import ${packageName}
112
+
113
+ nlp = ${packageName}.load() # or edsnlp.load("${packageName}")
114
+ `,
115
+ ];
116
+ };
117
+
100
118
  export const espnetTTS = (model: ModelData): string[] => [
101
119
  `from espnet2.bin.tts_inference import Text2Speech
102
120
 
@@ -133,6 +133,17 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = {
133
133
  repoName: "doctr",
134
134
  repoUrl: "https://github.com/mindee/doctr",
135
135
  },
136
+ edsnlp: {
137
+ prettyLabel: "EDS-NLP",
138
+ repoName: "edsnlp",
139
+ repoUrl: "https://github.com/aphp/edsnlp",
140
+ docsUrl: "https://aphp.github.io/edsnlp/latest/",
141
+ filter: false,
142
+ snippets: snippets.edsnlp,
143
+ countDownloads: {
144
+ wildcard: { path: "*/config.cfg" },
145
+ },
146
+ },
136
147
  elm: {
137
148
  prettyLabel: "ELM",
138
149
  repoName: "elm",
@@ -186,6 +197,13 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = {
186
197
  term: { path: "pytorch_model.bin" },
187
198
  },
188
199
  },
200
+ "gemma.cpp": {
201
+ prettyLabel: "gemma.cpp",
202
+ repoName: "gemma.cpp",
203
+ repoUrl: "https://github.com/google/gemma.cpp",
204
+ filter: false,
205
+ countDownloads: { wildcard: { path: "*.sbs" } },
206
+ },
189
207
  gliner: {
190
208
  prettyLabel: "GLiNER",
191
209
  repoName: "GLiNER",
@@ -307,6 +325,13 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = {
307
325
  snippets: snippets.pythae,
308
326
  filter: true,
309
327
  },
328
+ recurrentgemma: {
329
+ prettyLabel: "RecurrentGemma",
330
+ repoName: "recurrentgemma",
331
+ repoUrl: "https://github.com/google-deepmind/recurrentgemma",
332
+ filter: false,
333
+ countDownloads: { term: { path: "tokenizer.model" } },
334
+ },
310
335
  "sample-factory": {
311
336
  prettyLabel: "sample-factory",
312
337
  repoName: "sample-factory",
@@ -408,6 +433,13 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = {
408
433
  repoUrl: "https://github.com/TensorSpeech/TensorFlowTTS",
409
434
  snippets: snippets.tensorflowtts,
410
435
  },
436
+ "tic-clip": {
437
+ prettyLabel: "TiC-CLIP",
438
+ repoName: "TiC-CLIP",
439
+ repoUrl: "https://github.com/apple/ml-tic-clip",
440
+ filter: false,
441
+ countDownloads: { wildcard: { path: "checkpoints/*.pt" } },
442
+ },
411
443
  timesfm: {
412
444
  prettyLabel: "TimesFM",
413
445
  repoName: "timesfm",